結果

問題 No.3171 Color Restoration
コンテスト
ユーザー Tatsu_mr
提出日時 2025-06-06 22:33:59
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,321 bytes
コンパイル時間 4,221 ms
コンパイル使用メモリ 288,812 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-06-06 22:34:05
合計ジャッジ時間 5,276 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define For(i, a, b) for(int i = (a); i < (b); i++)
#define rep(i, n) For(i, 0, n)
#define rFor(i, a, b) for(int i = (a); i >= (b); i--)
#define ALL(v) (v).begin(), (v).end()
#define rALL(v) (v).rbegin(), (v).rend()
#define SZ(v) ((int)(v).size())

using lint = long long;
using ld = long double;

int INF = 2000000000;
lint LINF = 1000000000000000000;

struct SetupIo {
    SetupIo() {
        ios::sync_with_stdio(false);
        cin.tie(nullptr);
        cout << fixed << setprecision(15);
        cerr << fixed << setprecision(15);
    }
} setupio;

int main() {
    vector<set<string>> v = {
        {"gray","brown","green","cyan","blue","yellow","orange","red"},
        {"gray","green","blue","yellow","red"},
        {"gray","green","cyan","blue","violet","orange","red"}
    };
    string s1, s2, s3;
    cin >> s1 >> s2 >> s3;
    vector<int> id = {0, 1, 2};
    int cnt = 0;
    do {
        if (v[id[0]].count(s1) && v[id[1]].count(s2) && v[id[2]].count(s3)) {
            cnt++;
        }
    } while (next_permutation(ALL(id)));

    bool ok;
    if (s1 == s2 && s2 == s3) {
        ok = (cnt == 6);
    } else if (s1 == s2 || s2 == s3 || s3 == s1) {
        ok = (cnt == 2);
    } else {
        ok = (cnt == 1);
    }

    cout << (ok ? "Yes\n" : "No\n");
}
0