結果

問題 No.1618 Convolution?
ユーザー kokatsu
提出日時 2021-10-25 22:13:32
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 191 ms / 2,000 ms
コード長 870 bytes
コンパイル時間 2,484 ms
コンパイル使用メモリ 212,172 KB
実行使用メモリ 25,272 KB
最終ジャッジ日時 2024-06-22 12:59:09
合計ジャッジ時間 9,041 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std;

void main() {
    int N;
    readf("%d\n", N);

    auto A = readln.chomp.split.to!(long[]);
    auto B = readln.chomp.split.to!(long[]);

    auto X = A.cumulativeFold!"a + b".array;
    auto Y = B.cumulativeFold!"a + b".array;

    auto C = new long[](N*2); 

    auto S = new long[](N+1);
    auto T = new long[](N+1);
    foreach (i; 0 .. N) {
        S[i+1] = S[i] + X[i];
        T[i+1] = T[i] + Y[i];
        C[i+1] = S[i+1] + T[i+1];
    }

    auto U = new long[](N);
    auto V = new long[](N);
    U[0] = S[N];
    V[0] = T[N];
    foreach (i; 1 .. N) {
        U[i] = U[i-1] + X[N-1] - X[i-1] - N * A[i-1];
        V[i] = V[i-1] + Y[N-1] - Y[i-1] - N * B[i-1];
        C[N+i] = U[i] + V[i];
    }

    foreach (i, c; C) {
        if (i == N * 2 - 1) {
            writeln(c);
        }
        else {
            write(c, " ");
        }
    }
}
0