結果

問題 No.990 N×Mマス計算（Kの倍数）
コンテスト
ユーザー nanae
提出日時 2020-02-14 21:51:40
言語 D
(dmd 2.109.1)
結果
WA  
実行時間 -
コード長 1,846 bytes
コンパイル時間 726 ms
コンパイル使用メモリ 114,584 KB
実行使用メモリ 14,260 KB
最終ジャッジ日時 2024-06-22 05:09:00
合計ジャッジ時間 3,591 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 10 WA * 4 RE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

enum inf = 1_001_001_001;
enum infl = 1_001_001_001_001_001_001L;
enum mod = 1_000_000_007L;


void main() {
    int N, M; long K;
    scan(N, M, K);

    auto Bs = readln.split.array;
    bool tasu = Bs[0] == "+";
    Bs[0] = "0";
    auto B = Bs.to!(long[])[1 .. $].map!(bi => bi % K).array;

    auto A = iota(N).map!(i => readln.chomp.to!long % K).array;

    long ans;

    long[long] bc;
    foreach (bi ; B) {
        bc[bi]++;
    }

    if (tasu) {
        foreach (ai ; A) {
            if ((K - ai) % K in bc) ans += bc[(K - ai) % K];
        }
    }
    else {
        foreach (ai ; A) {
            if ((K / ai) % K in bc) ans += bc[(K / ai) % K];
            if (0 in bc) ans += bc[0];
        }
    }

    writeln(ans);
}






void scan(T...)(ref T args) {
    import std.stdio : readln;
    import std.algorithm : splitter;
    import std.conv : to;
    import std.range.primitives;

    auto line = readln().splitter();
    foreach (ref arg; args) {
        arg = line.front.to!(typeof(arg));
        line.popFront();
    }
    assert(line.empty);
}

void fillAll(R, T)(ref R arr, T value) {
    static if (is(typeof(arr[] = value))) {
        arr[] = value;
    }
    else {
        foreach (ref e; arr) {
            fillAll(e, value);
        }
    }
}

bool chmin(T, U...)(ref T x, U args) {
    bool isChanged;

    foreach (arg; args) {
        if (x > arg) {
            x = arg;
            isChanged = true;
        }
    }

    return isChanged;
}

bool chmax(T, U...)(ref T x, U args) {
    bool isChanged;

    foreach (arg; args) {
        if (x < arg) {
            x = arg;
            isChanged = true;
        }
    }

    return isChanged;
}
0