結果

問題 No.1424 Ultrapalindrome
コンテスト
ユーザー rym903
提出日時 2021-03-17 07:01:52
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,928 bytes
コンパイル時間 1,470 ms
コンパイル使用メモリ 131,592 KB
実行使用メモリ 12,288 KB
最終ジャッジ日時 2024-11-14 04:17:01
合計ジャッジ時間 3,837 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<algorithm>
#include<bitset>
#include<cmath>
#include<complex>
#include<deque>
#include<functional>
#include<iomanip>
#include<iostream>
#include<iterator>
#include<map>
#include<numeric>
#include<queue>
#include<set>
#include<stack>
#include<string>
#include<unordered_map>
#include<unordered_set>
#include<utility>
#include<vector>
#include<chrono>
using namespace std;
#define rep(i,n) for(int i=0; i<(n); i++)
#define FOR(i,x,n) for(int i=x; i<(n); i++)
#define vint(a,n) vint a(n); rep(i, n) cin >> a[i];
#define vll(a,n) vll a(n); rep(i, n) cin >> a[i];
#define ALL(n) begin(n),end(n)
#define RALL(n) rbegin(n),rend(n)
#define MOD (1000000007)
// #define MOD (998244353)
#define INF (1e9+7)
#define INFL (2e18)

typedef long long ll;
typedef unsigned int ui;
typedef unsigned long long ull;
using vint=vector<int>;
using vll=vector<ll>;
using vbool=vector<bool>;
using P=pair<ll,ll>;
template<class T>using arr=vector<vector<T>>;
template<class T>int popcount(T &a){int c=0; rep(i, 8*(int)sizeof(a)){if((a>>i)&1) c++;} return c;}
template<class T>void pl(T x){cout << x << " ";}
template<class T>void pr(T x){cout << x << endl;}
template <class T, class... Ts> inline void pr(T Tar, Ts... ts) { std::cout << Tar << " "; pr(ts...); return; }
template<class T>void prvec(vector<T>& a){rep(i, (int)a.size()-1){pl(a[i]);} pr(a.back());}
template<class T>void prarr(arr<T>& a){rep(i, (int)a.size()) if(a[i].empty()) pr(""); else prvec(a[i]);}
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }
template<typename A, size_t N, typename T> void Fill(A (&array)[N], const T &val){fill( (T*)array, (T*)(array+N), val );}
// dijkstra法。ただし、全ての枝の距離が1のとき、BFS
void dijkstra(int st, arr<P>& G, vll& dist, vbool& seen, bool is_all_one=false){
    int V = G.size();
    priority_queue<P, vector<P>, greater<P>> min_heap;
    min_heap.push({0, st});
    dist.assign(V, INFL);
    seen.assign(V, false);
    dist[st] = 0;

    // 全ての枝の距離が1なら、単純なbfsで求められる。
    if(is_all_one){
        queue<int> que;
        que.push(st);
        seen[st] = true;
        while(!que.empty()){
            int now = que.front(); que.pop();
            ll d = dist[now];
            for(auto p: G[now]){
                int nx = p.second;
                if(seen[nx]) continue;
                seen[nx] = true;
                dist[nx] = d + 1;
                que.push(nx);
            }
        }
        return;
    }
 
    int frm;
    ll d;
    while(!min_heap.empty()){
        tie(d, frm) = min_heap.top(); min_heap.pop();
        if(seen[frm]) continue;
 
        // printf("node=%d", frm);
        seen[frm] = true;
        dist[frm] = d;
 
        for(P e: G[frm]){
            if(dist[e.second] > d+e.first)
                min_heap.push({d+e.first, e.second});
                // printf("(%d, %d) pushed", d+e.first, e.second);
        }
    }
}
int main()
{
    int n; cin >> n;
    arr<P> g(n);
    vint in(n, 0);
    int st;
    rep(i, n-1){
        int a, b; cin >> a >> b;
        a--; b--;
        g[a].push_back({1, b});
        g[b].push_back({1, a});
        in[a]++;
        in[b]++;
    }
    vint v;
    rep(i, n) {
        if(in[i]==1) v.push_back(i);
    }
    // if(v.size()==2){
    //     pr("Yes");
    //     return 0;
    // }
    st = v[0];

    vll dist;
    vbool seen;
    dijkstra(st, g, dist, seen, 1);

    // pr(st);
    // prvec(v);
    // prvec(dist);

    rep(i, v.size()-2){
        if(dist[v[i+1]] != dist[v[i+2]]){
            pr("No");
            return 0;
        }
    }

    swap(v[0], v[1]);
    st = v[0];
    dijkstra(st, g, dist, seen, 1);
    rep(i, v.size()-2){
        if(dist[v[i+1]] != dist[v[i+2]]){
            pr("No");
            return 0;
        }
    }
    pr("Yes");
    return 0;}
0