所谓的日常 #8

div.2

CodeForces 475A Bayan Bus

题意就是安排K个人的座位并且输出咯。

嘛，这是我以前的代码...好像比大家都长哈哈哈...

#include <bits/stdc++.h>

int n;
int A[4][50];

void paint() {
    for (int i = 0; n; ++ i) {
        if (i == 0) {
            int c = std::min(4,n);
            for (int j = 0; j < c; ++ j) {
                A[j][i] = 1;
            }
            n -= c;
        } else {
            int c = std::min(3,n);
            for (int j = 0; j < c; ++ j) {
                A[j][i] = 1;
            }
            std::swap(A[3][i],A[2][i]);
            n -= c;
        }
    }
    putchar('+');
    for (int i = 0; i < 12; ++ i) {
        printf("--");
    }
    putchar('+');
    puts("");
    for (int i = 0; i < 4; ++ i) {
        putchar('|');
        for (int j = 0; j < 11; ++ j) {
            if (i==2) {
                if (A[i][j]) putchar('O');
                else if (j==0) putchar('#');
                else putchar('.');
            } else {
                if (A[i][j]) putchar('O');
                else putchar('#');
            }
            putchar('.');
        }
        if (i==2) putchar('.');
        else putchar('|');
        if (i==0) putchar('D');
        else putchar('.');
        if (i==0 || i==3) puts("|)");
        else puts("|");
    }
    putchar('+');
    for (int i = 0; i < 12; ++ i) {
        printf("--");
    }
    putchar('+');
    puts("");
}

int main() {
    scanf("%d",&n);
    paint();
    return 0;
}

div.1

CodeForces 466C Number of Ways

把长度为n(<=5e5)的数组分成三段非空段且和相等 的方案数。

所以枚举第二段末尾位置，拿一个cnt记录之前有多少满足条件的第一段末尾 就好啦。

#include <bits/stdc++.h>
typedef long long LL;

const int N = 500000 + 5;
int A[N],n;

LL work() {
    LL sum = std::accumulate(A,A + n,0ll);
    if (sum % 3 != 0) return 0;

    LL now = 0,ret = 0,cnt = 0;
    for (int i = 0; i < n - 1; ++ i) {
        now += A[i];
        if (now == sum / 3 * 2) {
            ret += cnt;
        }
        if (now == sum / 3) {
            cnt ++;
        }
    }
    return ret;
}

int main() {
    scanf("%d",&n);
    for (int i = 0; i < n; ++ i) {
        scanf("%d",A + i);
    }
    printf("%I64d
",work());
}