結果

問題 No.1470 Mex Sum
ユーザー kokatsukokatsu
提出日時 2021-10-18 00:12:59
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 30 ms / 2,000 ms
コード長 670 bytes
コンパイル時間 2,436 ms
コンパイル使用メモリ 212,000 KB
実行使用メモリ 10,832 KB
最終ジャッジ日時 2024-06-22 12:49:50
合計ジャッジ時間 5,474 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 4 ms
6,944 KB
testcase_03 AC 3 ms
6,944 KB
testcase_04 AC 3 ms
6,940 KB
testcase_05 AC 3 ms
6,940 KB
testcase_06 AC 4 ms
6,944 KB
testcase_07 AC 4 ms
6,940 KB
testcase_08 AC 4 ms
6,940 KB
testcase_09 AC 4 ms
6,944 KB
testcase_10 AC 4 ms
6,944 KB
testcase_11 AC 3 ms
6,944 KB
testcase_12 AC 28 ms
9,812 KB
testcase_13 AC 28 ms
9,912 KB
testcase_14 AC 28 ms
8,752 KB
testcase_15 AC 27 ms
10,832 KB
testcase_16 AC 28 ms
9,676 KB
testcase_17 AC 27 ms
9,180 KB
testcase_18 AC 26 ms
10,080 KB
testcase_19 AC 26 ms
9,852 KB
testcase_20 AC 26 ms
9,036 KB
testcase_21 AC 28 ms
9,980 KB
testcase_22 AC 29 ms
9,860 KB
testcase_23 AC 28 ms
10,212 KB
testcase_24 AC 29 ms
9,440 KB
testcase_25 AC 28 ms
8,972 KB
testcase_26 AC 29 ms
9,240 KB
testcase_27 AC 29 ms
10,144 KB
testcase_28 AC 29 ms
9,212 KB
testcase_29 AC 30 ms
10,116 KB
testcase_30 AC 29 ms
10,024 KB
testcase_31 AC 28 ms
9,788 KB
testcase_32 AC 29 ms
9,696 KB
testcase_33 AC 30 ms
10,392 KB
testcase_34 AC 29 ms
10,360 KB
testcase_35 AC 28 ms
9,368 KB
testcase_36 AC 28 ms
9,764 KB
testcase_37 AC 22 ms
9,048 KB
testcase_38 AC 22 ms
9,656 KB
testcase_39 AC 22 ms
9,480 KB
testcase_40 AC 23 ms
9,528 KB
testcase_41 AC 24 ms
8,760 KB
testcase_42 AC 23 ms
8,996 KB
testcase_43 AC 24 ms
9,060 KB
testcase_44 AC 24 ms
10,000 KB
testcase_45 AC 24 ms
10,584 KB
testcase_46 AC 23 ms
8,716 KB
testcase_47 AC 22 ms
9,364 KB
testcase_48 AC 24 ms
9,568 KB
testcase_49 AC 23 ms
9,488 KB
testcase_50 AC 22 ms
8,856 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std;

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

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

    auto one = iota(0, N).map!(a => A[a] == 1).cumulativeFold!"a + b".array;
    auto two = iota(0, N).map!(a => A[a] == 2).cumulativeFold!"a + b".array;

    long res;
    foreach (i; 0 .. N-1) {
        int rem = N - i - 1;
        int cnt1 = one[N-1] - one[i];
        int cnt2 = two[N-1] - two[i];
        if (A[i] == 1) {
            res += cnt2 * 3 + (rem - cnt2) * 2; 
        }
        else if (A[i] == 2) {
            res += cnt1 * 3 + (rem - cnt1);
        }
        else {
            res += cnt1 * 2 + (rem - cnt1);
        }
    }

    res.writeln;
}
0