結果

問題 No.2343 (l+r)/2
コンテスト
ユーザー Syed Shakil Mahmud (Shajib)
提出日時 2023-06-09 21:10:06
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,179 bytes
コンパイル時間 796 ms
コンパイル使用メモリ 67,960 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2025-01-02 01:52:02
合計ジャッジ時間 2,254 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1
other AC * 5 WA * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>

std::string isPossible(std::vector<int>& A) {
    int n = A.size();
    int countZeros = 0;
    int countOnes = 0;

    for (int i = 0; i < n; i++) {
        if (A[i] == 0) {
            countZeros++;
        } else {
            countOnes++;
        }
    }

    // 最初の要素が0でない場合、不可能
    if (A[0] != 0) {
        return "No";
    }

    // 全ての要素が0の場合、不可能
    if (countOnes == 0) {
        return "No";
    }

    // 操作によって(0.5)になるか判定
    for (int i = 1; i < n; i++) {
        if (A[i] == 0) {
            if (countOnes < 2) {
                return "No";
            }
            countOnes--;
        } else {
            if (countZeros < 1) {
                return "No";
            }
            countZeros--;
            countOnes++;
        }
    }

    return "Yes";
}

int main() {
    int T;
    std::cin >> T;

    while (T--) {
        int N;
        std::cin >> N;

        std::vector<int> A(N);
        for (int i = 0; i < N; i++) {
            std::cin >> A[i];
        }

        std::cout << isPossible(A) << std::endl;
    }

    return 0;
}
0