結果

問題 No.1424 Ultrapalindrome
ユーザー rym903rym903
提出日時 2021-03-17 07:01:52
言語 C++14
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,820 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 2 ms
6,820 KB
testcase_05 AC 2 ms
6,820 KB
testcase_06 AC 2 ms
6,820 KB
testcase_07 AC 1 ms
6,820 KB
testcase_08 WA -
testcase_09 AC 74 ms
10,112 KB
testcase_10 AC 77 ms
10,240 KB
testcase_11 AC 52 ms
8,320 KB
testcase_12 AC 74 ms
9,728 KB
testcase_13 AC 18 ms
6,816 KB
testcase_14 AC 59 ms
8,704 KB
testcase_15 AC 2 ms
6,816 KB
testcase_16 AC 46 ms
7,552 KB
testcase_17 AC 56 ms
8,448 KB
testcase_18 AC 38 ms
6,820 KB
testcase_19 AC 61 ms
8,448 KB
testcase_20 AC 14 ms
6,820 KB
testcase_21 AC 47 ms
7,296 KB
testcase_22 AC 29 ms
6,816 KB
testcase_23 AC 7 ms
6,816 KB
testcase_24 AC 13 ms
6,816 KB
testcase_25 AC 11 ms
6,820 KB
testcase_26 AC 74 ms
9,344 KB
testcase_27 WA -
testcase_28 AC 74 ms
11,392 KB
testcase_29 AC 74 ms
11,392 KB
testcase_30 AC 62 ms
12,288 KB
testcase_31 AC 73 ms
12,288 KB
権限があれば一括ダウンロードができます

ソースコード

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