結果

問題 No.2860 Heal Slimes
コンテスト
ユーザー dyktr_06
提出日時 2025-04-18 16:15:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,561 bytes
コンパイル時間 750 ms
コンパイル使用メモリ 73,140 KB
実行使用メモリ 6,528 KB
最終ジャッジ日時 2025-04-18 16:15:48
合計ジャッジ時間 4,120 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 50 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <numeric> // 標準ライブラリインクルード（今回は直接は使わない）

// gemini

// 負の数も扱えるように剰余を計算する関数 (modulus > 0 を想定)
long long correct_modulo(long long value, long long modulus) {
    long long result = value % modulus;
    // 結果が負の場合は、modulus を足して非負にする
    return result < 0 ? result + modulus : result;
}

// 各テストケースを処理する関数
void solve() {
    long long n; // スライムの数
    long long k; // 回復魔法の対象となる連続するスライムの数
    long long x; // 回復量
    std::cin >> n >> k >> x;

    std::vector<long long> h(n); // 各スライムの体力
    for (int i = 0; i < n; ++i) {
        std::cin >> h[i];
    }

    // --- 条件1: 全てのスライムの体力が X で割った余りが等しいか ---
    long long h0_mod = correct_modulo(h[0], x); // 基準となる余り
    for (int i = 1; i < n; ++i) {
        if (correct_modulo(h[i], x) != h0_mod) {
            std::cout << "No\n"; // 結果を出力してこのテストケースの処理を終了
            return;
        }
    }

    // --- 条件2: 差分に基づく累積和の非正性チェック ---
    // q_i = (Hi - H{i-1}) / X を計算 (i = 1, ..., N-1)
    std::vector<long long> q(n - 1);
    for (int i = 1; i < n; ++i) {
        q[i - 1] = (h[i] - h[i - 1]) / x;
    }

    // グループ m = 1 から K-1 について、累積和をチェック
    for (int m = 1; m < k; ++m) {
        long long current_prefix_sum = 0; // 現在の累積和
        // インデックス i がグループ m に属するものを順に見ていく (i = m, m+K, m+2K, ...)
        for (int i = m; i < n; i += k) {
            // q 配列のインデックスは i-1
            current_prefix_sum += q[i - 1];
            if (current_prefix_sum > 0) {
                // 累積和が一度でも正になったら不可能
                std::cout << "No\n"; // 結果を出力してこのテストケースの処理を終了
                return;
            }
        }
    }

    // 全ての条件を満たせば可能
    std::cout << "Yes\n"; // 結果を出力
}

int main() {
    // 高速化のための設定
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(NULL);

    int t; // テストケース数
    std::cin >> t;
    while (t--) { // T 回ループ
        solve(); // 各テストケースを解く関数を呼び出す
    }
    return 0;
}
0