結果

問題 No.1017 Reiwa Sequence
コンテスト
ユーザー 👑 tttmmmyyyy
提出日時 2026-11-04 20:42:02
言語 Fix
(1.6.0 + cp-library + minilib)
コンパイル:
fix_c _filename_
実行:
./a.out
結果
WA  
実行時間 -
コード長 1,789 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 18,149 ms
コンパイル使用メモリ 670,124 KB
実行使用メモリ 18,796 KB
最終ジャッジ日時 2026-11-04 20:42:53
合計ジャッジ時間 45,619 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge4_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 37 WA * 14
権限があれば一括ダウンロードができます
コンパイルメッセージ
make: 'lib.o' は更新済みです.

ソースコード

diff #
raw source code 

module Main;

import CPLib.IO;

// import BinaryHeap;
// import BoolArray;
// import CPLib.Arithmetic;
// import CPLib.Bipartite;
import CPLib.Bits;
// import CPLib.Convolution;
// import CPLib.DSegtree;
// import CPLib.Graph;
// import CPLib.LDSegtree;
// import CPLib.LSegtree;
// import CPLib.MaxFlow;
// import CPLib.Memoize;
// import CPLib.MinCostFlow;
// import CPLib.Misc;
// import CPLib.Monoid;
// import CPLib.Segtree;
// import CPLib.String;
// import CPLib.Trait;
// import CPLib.TwoSat;
// import CPLib.UnionFind;
// import CPLib.ZP;
import HashMap;
// import HashSet;

cM : I64 = 150000;

main : IO () = (
    let iN = *read_i;
    let iAs = *read_is(iN);
    let cI = (
        count_up(1).filter(|n|
            1.shift_left(n) > cM
        ).get_first.as_some
    );
    let cL = min(cI, iN);
    let mp = HashMap::empty(1.shift_left(cL));
    let res = range(0, 1.shift_left(cL)).loop_iter_s(mp, |flag, mp|
        let sum = range(0, cL).filter(|bit| flag.bit_check(bit)).map(|i| iAs.@(i)).sum;
        match mp.find(sum) {
            none() => (
                mp.insert(sum, flag).continue
            ),
            some(flag1) => (
                (flag, flag1).break
            )
        }
    );
    match res {
        continue(_) => (
            println("No")
        ),
        break((flag0, flag1)) => (
            let xs = range(0, cL).map(|bit|
                if flag0.bit_check(bit) && flag1.bit_check(bit) { 0 };
                if flag0.bit_check(bit) { 1 };
                if flag1.bit_check(bit) { -1 };
                0
            ).to_array;
            let ans = range(0, cL).map(|i| iAs.@(i) * xs.@(i)).to_array + fill(iN - cL, 0);
            println("Yes");;
            "{}".printfln((ans.to_iter.joined_with(" "),))
        ),
    }
);
0