結果

問題 No.2518 Adjacent Larger
ユーザー InTheBloom
提出日時 2023-10-27 22:34:09
言語 D
(dmd 2.109.1)
結果
WA  
実行時間 -
コード長 1,404 bytes
コンパイル時間 2,704 ms
コンパイル使用メモリ 177,320 KB
実行使用メモリ 7,096 KB
最終ジャッジ日時 2024-09-25 14:36:44
合計ジャッジ時間 4,083 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 14 WA * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std;

void main () {
    int T = readln.chomp.to!int;

    foreach (_; 0..T) {
        int N = readln.split[0].to!int;
        int[] A = readln.split.to!(int[]);

        auto ans = solve(N, A);
        writeln(ans);
    }
}

auto solve (int N, int[] A) {
    /**
     * 必要条件
     * - 0と2が少なくとも1つ
     * - 0と2は同じものが連続することがない
     * - 0と2は同じ数だけ存在
     * かなー未証明だけど提出チャレンジする
     * 
     * - 0の次は2が来ないとダメっていうの忘れてた
     */

    int zero = 0, two = 0;
    foreach (a; A) {
        if (a == 0) zero++;
        if (a == 2) two++;
    }

    if (zero != two || zero == 0 || two == 0) return "No";

    foreach (i; 0..A.length) {
        if (A[i] == 0 || A[i] == 2) {
            if (A[i] == A[(i+1)%A.length]) return "No";
        }
    }

    bool isNextZero = false;
    bool first = true;
    foreach (a; chain(A, A)) {
        if (first && a == 0) { isNextZero = true; first = false; }
        if (first && a == 2) { isNextZero = false; first = false; }

        if (first) continue;

        if (a == 0 && !isNextZero) {
            return "No";
        } else {
            isNextZero = false;
        }

        if (a == 2 && isNextZero) {
            return "No";
        } else {
            isNextZero = true;
        }
    }

    return "Yes";
}
0