結果

問題 No.2408 Lakes and Fish
ユーザー InTheBloomInTheBloom
提出日時 2023-08-11 22:06:42
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 150 ms / 2,000 ms
コード長 1,579 bytes
コンパイル時間 4,203 ms
コンパイル使用メモリ 179,384 KB
実行使用メモリ 18,672 KB
最終ジャッジ日時 2024-11-18 16:29:11
合計ジャッジ時間 7,041 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 1 ms
6,820 KB
testcase_02 AC 1 ms
6,820 KB
testcase_03 AC 1 ms
6,816 KB
testcase_04 AC 131 ms
18,288 KB
testcase_05 AC 133 ms
17,468 KB
testcase_06 AC 122 ms
18,672 KB
testcase_07 AC 149 ms
12,980 KB
testcase_08 AC 149 ms
13,048 KB
testcase_09 AC 149 ms
12,472 KB
testcase_10 AC 150 ms
13,040 KB
testcase_11 AC 77 ms
12,004 KB
testcase_12 AC 107 ms
17,920 KB
testcase_13 AC 29 ms
6,820 KB
testcase_14 AC 61 ms
9,032 KB
testcase_15 AC 122 ms
17,488 KB
testcase_16 AC 41 ms
7,608 KB
testcase_17 AC 32 ms
6,816 KB
testcase_18 AC 71 ms
12,120 KB
testcase_19 AC 115 ms
17,292 KB
testcase_20 AC 108 ms
12,368 KB
testcase_21 AC 119 ms
17,364 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std;

struct trio {
    int F;
    int B;
    int W;
}

void main () {
    int N, M; readln.read(N, M);
    int[] L = readln.split.to!(int[]);

    trio[] fish = new trio[](M);
    foreach (i; 0..M) {
        int F, B, W; readln.read(F, B, W);
        fish[i] = trio(F, B, W);
    }

    solve(N, M, L, fish);
}

void solve (int N, int M, int[] L, trio[] fish) {
    // ある魚について、水場に動かすべきかどうかを貪欲に取れば良い
    L.sort;
    bool[int] water;
    foreach (l; L) {
        water[l] = true;
    }

    long ans = 0;
    foreach (f; fish) {
        // すでに水の中にいるか？
        if (f.F in water) {
            ans += f.W;
            continue;
        }

        // 一番近い水場はどこか？
        int near;
        if (f.F < L[0]) {
            near = L[0];
        } else if (L[$-1] < f.F) {
            near = L[$-1];
        } else {
            int ok = 0, ng = cast(int) (L.length-1);
            while (1 < abs(ok-ng)) {
                int mid = (ok+ng) / 2;
                if (L[mid] < f.F) {
                    ok = mid;
                } else {
                    ng = mid;
                }
            }
            if (abs(f.F - L[ok]) < abs(f.F - L[ng])) {
                near = L[ok];
            } else {
                near = L[ng];
            }
        }

        ans += max(f.B, f.W - abs(near - f.F));
    }

    writeln(ans);
}

void read(T...)(string S, ref T args) {
    auto buf = S.split;
    foreach (i, ref arg; args) {
        arg = buf[i].to!(typeof(arg));
    }
}
0