結果

問題 No.1424 Ultrapalindrome
ユーザー dekomori_sanaedekomori_sanae
提出日時 2021-03-12 22:19:23
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,396 bytes
コンパイル時間 1,005 ms
コンパイル使用メモリ 96,848 KB
実行使用メモリ 14,212 KB
最終ジャッジ日時 2024-04-22 13:20:19
合計ジャッジ時間 3,332 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 WA -
testcase_09 AC 82 ms
11,520 KB
testcase_10 AC 81 ms
11,520 KB
testcase_11 AC 58 ms
9,216 KB
testcase_12 AC 74 ms
11,136 KB
testcase_13 AC 20 ms
5,376 KB
testcase_14 AC 61 ms
9,600 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 47 ms
8,320 KB
testcase_17 AC 56 ms
9,344 KB
testcase_18 AC 36 ms
7,424 KB
testcase_19 AC 59 ms
9,472 KB
testcase_20 AC 14 ms
5,376 KB
testcase_21 AC 46 ms
8,192 KB
testcase_22 AC 29 ms
6,528 KB
testcase_23 AC 7 ms
5,376 KB
testcase_24 AC 13 ms
5,376 KB
testcase_25 AC 12 ms
5,376 KB
testcase_26 AC 72 ms
10,496 KB
testcase_27 WA -
testcase_28 AC 77 ms
13,440 KB
testcase_29 AC 76 ms
13,568 KB
testcase_30 AC 64 ms
14,212 KB
testcase_31 AC 76 ms
14,212 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdio>
#include <string>
#include <algorithm>
#include <utility>
#include <cmath>
#include <vector>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <map>
#include <tuple>
#include <numeric>
#include <functional>
using namespace std;
typedef long long ll;
typedef vector<ll> vl;
typedef vector<vector<ll>> vvl;
typedef pair<ll, ll> P;
#define rep(i, n) for(ll i = 0; i < n; i++)
#define exrep(i, a, b) for(ll i = a; i <= b; i++)
#define out(x) cout << x << endl
#define exout(x) printf("%.10f\n", x)
#define chmax(x, y) x = max(x, y)
#define chmin(x, y) x = min(x, y)
#define all(a) a.begin(), a.end()
#define rall(a) a.rbegin(), a.rend()
#define pb push_back
#define re0 return 0
const ll mod = 1000000007;
const ll INF = 1e16;

struct Edge {
    ll to, cost;
    Edge(ll to, ll cost): to(to), cost(cost) {}
};

int main() {
    ll n;
    cin >> n;
    
    vector<vector<Edge>> G(n);
    vl deg(n);
    rep(i, n-1) {
        ll a, b;
        cin >> a >> b;
        a--;  b--;
        G[a].emplace_back(b, 1);
        G[b].emplace_back(a, 1);
        deg[a]++;  deg[b]++;
    }

    if(n == 1) {
        out("Yes");
        re0;
    }

    ll root = -1;
    ll ma = 0;
    rep(v, n) {
        if(ma < deg[v]) {
            root = v;
            ma = deg[v];
        }
    }
    
    vl dist(n, -1);  // dist[v] : 頂点vの根からの距離
    queue<ll> Q;
    dist[root] = 0;
    Q.push(root);
    while(!Q.empty()) {
        ll v = Q.front();  Q.pop();
        for(Edge e : G[v]) {
            if(dist[e.to] == -1) {
                dist[e.to] = dist[v] + e.cost;
                Q.push(e.to);
            }
        }
    }

    vl dist1(n, -1);  // dist1[v] : 頂点vの頂点0からの距離
    queue<ll> Q1; 
    dist1[0] = 0;
    Q1.push(0);
    while(!Q1.empty()) {
        ll v = Q1.front();
        Q1.pop();
        for(auto p : G[v]) {
            ll u = p.to;
            ll cost = p.cost; 
            if(dist1[u] != -1) {
                continue;
            }
            dist1[u] = dist1[v] + cost;
            Q1.push(u);
        }
    }
    
    ll vertex1 = 0;  // 頂点0からの距離が最大の頂点
    rep(v, n) {
        if(dist1[vertex1] < dist1[v]) {
            vertex1 = v;
        }
    }
    
    vl dist2(n, -1);  // dist2[v] : 頂点vの頂点vertex1からの距離
    queue<ll> Q2;
    dist2[vertex1] = 0;
    Q2.push(vertex1);
    while(!Q2.empty()) {
        ll v = Q2.front();
        Q2.pop();
        for(auto p : G[v]) {
            ll u = p.to;
            ll cost = p.cost; 
            if(dist2[u] != -1) {
                continue;
            }
            dist2[u] = dist2[v] + cost;
            Q2.push(u);
        }
    }
 
    ll vertex2 = vertex1;  // 頂点vertex1からの距離が最大の頂点
    ll TreeDiameter = 0;  // 木の直径
    rep(v, n) {
        if(dist2[vertex2] < dist2[v]) {
            vertex2 = v;
            TreeDiameter = dist2[v];
        }
    }
    
    bool ok = true;
    ll x = -1;
    rep(v, n) {
        if(v != root && deg[v] == 1) {
            if(x == -1) {
                x = dist[v];
            }
            else if(x != dist[v]) {
                ok = false;
            }
        }
    }

    if(TreeDiameter == n-1) {
        ok = true;
    }

    if(ok) {
        out("Yes");
    }
    else {
        out("No");
    }
    
    re0;
}
0