結果

問題 No.1424 Ultrapalindrome
コンテスト
ユーザー ぷら
提出日時 2021-03-12 22:44:29
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,039 bytes
コンパイル時間 1,902 ms
コンパイル使用メモリ 181,380 KB
実行使用メモリ 10,676 KB
最終ジャッジ日時 2024-10-14 13:18:10
合計ジャッジ時間 4,076 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20 WA * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int main() {
    int N;
    cin >> N;
    vector<vector<int>>ki(N);
    for(int i = 0; i < N-1; i++) {
        int a,b;
        cin >> a >> b;
        a--;
        b--;
        ki[a].push_back(b);
        ki[b].push_back(a);
    }
    vector<int>tmp;
    for(int i = 0; i < N; i++) {
        if(ki[i].size() == 1) {
            tmp.push_back(i);
        }
    }
    if(tmp.size() == 2) {
        cout << "Yes" << endl;
        return 0;
    }
    vector<int>dist(N,-1);
    dist[tmp[0]] = 0;
    queue<int>que;
    que.push(tmp[0]);
    while(!que.empty()) {
        int x = que.front();
        que.pop();
        for(int i = 0; i < ki[x].size(); i++) {
            if(dist[ki[x][i]] == -1) {
                dist[ki[x][i]] = dist[x]+1;
                que.push(ki[x][i]);
            }
        }
    }
    if(dist[tmp[1]]%2 == 1) {
        cout << "No" << endl;
    }
    else {
        vector<int>tmp2;
        tmp2.push_back(tmp[1]);
        que.push(tmp[1]);
        while(!que.empty()) {
            int x = que.front();
            que.pop();
            for(int i = 0; i < ki[x].size(); i++) {
                if(dist[ki[x][i]] == dist[x]-1) {
                    tmp2.push_back(ki[x][i]);
                    que.push(ki[x][i]);
                    que.pop();
                }
            }
        }
        int p = tmp2[tmp2.size()/2];
        vector<int>dist2(N,-1);
        dist2[p] = 0;
        que.push(p);
        while(!que.empty()) {
            int x = que.front();
            que.pop();
            for(int i = 0; i < ki[x].size(); i++) {
                if(dist2[ki[x][i]] == -1) {
                    dist2[ki[x][i]] = dist2[x]+1;
                    que.push(ki[x][i]);
                }
            }
        }
        set<int>st;
        for(int i = 0; i < tmp.size(); i++) {
            st.insert(dist2[tmp[i]]);
        }
        if(st.size() == 1) {
            cout << "Yes" << endl;
        }
        else {
            cout << "No" << endl;
        }
    }
}
0