結果

問題 No.619 CardShuffle
ユーザー nebukuro09nebukuro09
提出日時 2018-01-12 17:31:42
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 2,661 ms / 3,000 ms
コード長 3,422 bytes
コンパイル時間 922 ms
コンパイル使用メモリ 119,808 KB
実行使用メモリ 120,064 KB
最終ジャッジ日時 2024-06-12 23:32:20
合計ジャッジ時間 31,114 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 AC 1 ms
5,376 KB
testcase_12 AC 1 ms
5,376 KB
testcase_13 AC 1 ms
5,376 KB
testcase_14 AC 1 ms
5,376 KB
testcase_15 AC 1 ms
5,376 KB
testcase_16 AC 1,524 ms
105,728 KB
testcase_17 AC 2,027 ms
120,064 KB
testcase_18 AC 1,532 ms
102,400 KB
testcase_19 AC 2,016 ms
119,936 KB
testcase_20 AC 895 ms
50,176 KB
testcase_21 AC 2,064 ms
119,936 KB
testcase_22 AC 2,650 ms
120,064 KB
testcase_23 AC 2,035 ms
120,064 KB
testcase_24 AC 2,661 ms
119,936 KB
testcase_25 AC 1,026 ms
70,272 KB
testcase_26 AC 1,127 ms
70,272 KB
testcase_27 AC 1,407 ms
79,232 KB
testcase_28 AC 1,154 ms
70,272 KB
testcase_29 AC 1,431 ms
79,488 KB
testcase_30 AC 751 ms
50,176 KB
testcase_31 AC 1 ms
5,376 KB
testcase_32 AC 902 ms
50,176 KB
testcase_33 AC 1,471 ms
103,936 KB
testcase_34 AC 1,050 ms
70,272 KB
testcase_35 AC 224 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.stdio, std.array, std.string, std.conv, std.algorithm;
import std.typecons, std.range, std.random, std.math, std.container;
import std.numeric, std.bigint, core.bitop;

immutable long MOD = 10^^9 + 7;

void main() {
    auto N = readln.chomp.to!int;
    auto C = readln.split;
    auto A = iota(0, N, 2).map!(i => C[i].to!long).array;
    auto B = iota(1, N, 2).map!(i => C[i]).array;

    auto st = new SegmentTree(N/2+1, B.dup);
    foreach (i; 0..N/2+1) st.update(i, A[i]);

    auto Q = readln.chomp.to!int;
    while (Q--) {
        auto q = readln.split;
        auto a = q[1].to!int - 1;
        auto b = q[2].to!int - 1;
        auto parity = a % 2;
        a /= 2;
        b /= 2;
        if (q[0] == "?") {
            st.query(a, b).writeln;
        } else if (parity) {
            swap(st.ops[a], st.ops[b]);
            st.update(a, A[a]);
            st.update(a+1, A[a+1]);
            st.update(b, A[b]);
            st.update(b+1, A[b+1]);
        } else {
            swap(A[a], A[b]);
            st.update(a, A[a]);
            st.update(b, A[b]);
        }
    }

}

class SegmentTree {
    long[][] table;
    string[] ops;
    int size;

    this(int n, string[] ops) {
        assert(bsr(n) < 29);
        size = 1 << (bsr(n) + 2);
        table = new long[][](size);
        this.ops = ops;
    }

    long[] merge(int mid, long[] a1, long[] a2) {
        long[] ret;
        if (a1 == [] || a2 == []) {
            ret = a1 ~ a2;
        } else if (ops[mid] == "+") {
            long tmp = 0;
            foreach (x; a1[1..$]) (tmp += x) %= MOD;
            foreach (x; a2[0..$-1]) (tmp += x) %= MOD;
            ret = [];
            ret ~= a1.front;
            if (tmp > 0) ret ~= tmp;
            ret ~= a2.back;
        } else {
            long tmp = a1.back * a2.front % MOD;
            if (a1.length == 1 && a2.length == 1) {
                ret = [tmp];
            } else if (a1.length == 1) {
                ret = tmp ~ a2[1..$];
            } else if (a2.length == 1) {
                ret = a1[0..$-1] ~ tmp;
            } else {
                foreach (x; a1[1..$-1]) (tmp += x) %= MOD;
                foreach (x; a2[1..$-1]) (tmp += x) %= MOD;
                ret = [a1.front, tmp, a2.back];
            }
        }
        return ret;
    }
    
    void update(int pos, long num) {
        return update(pos, num, 0, 0, size/2-1);
    }

    void update(int pos, long num, int i, int left, int right) {
        if (left == right) {
            table[i] = [num];
            return;
        }
        
        auto mid = (left + right) / 2;
        if (pos <= mid)
            update(pos, num, i*2+1, left, mid);
        else
            update(pos, num, i*2+2, mid+1, right);

        table[i] = merge(mid, table[i*2+1], table[i*2+2]);
    }

    long query(int pl, int pr) {
        auto X = query(pl, pr, 0, 0, size/2-1);
        long ret = 0;
        foreach (x; X) (ret += x) %= MOD;
        return ret;
    }

    long[] query(int pl, int pr, int i, int left, int right) {
        if (pl > right || pr < left) {
            return [];
        } else if (pl <= left && right <= pr) {
            return table[i];
        } else {
            auto a = query(pl, pr, i*2+1, left, (left+right)/2);
            auto b = query(pl, pr, i*2+2, (left+right)/2+1, right);
            auto x =  merge((left + right) / 2, a, b);
            return x;
        }
    }
}
0