結果

問題 No.1424 Ultrapalindrome
ユーザー siro53siro53
提出日時 2021-03-12 22:12:22
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,253 bytes
コンパイル時間 2,910 ms
コンパイル使用メモリ 207,468 KB
実行使用メモリ 12,928 KB
最終ジャッジ日時 2024-04-22 13:14:02
合計ジャッジ時間 3,822 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 WA -
testcase_06 AC 1 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 WA -
testcase_09 AC 40 ms
8,372 KB
testcase_10 AC 40 ms
8,268 KB
testcase_11 AC 28 ms
6,776 KB
testcase_12 AC 37 ms
8,004 KB
testcase_13 AC 10 ms
5,376 KB
testcase_14 AC 30 ms
7,060 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 25 ms
6,328 KB
testcase_17 AC 29 ms
6,848 KB
testcase_18 AC 16 ms
5,692 KB
testcase_19 AC 32 ms
6,784 KB
testcase_20 AC 8 ms
5,376 KB
testcase_21 AC 26 ms
6,144 KB
testcase_22 AC 15 ms
5,376 KB
testcase_23 AC 4 ms
5,376 KB
testcase_24 AC 7 ms
5,376 KB
testcase_25 AC 7 ms
5,376 KB
testcase_26 AC 36 ms
7,424 KB
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 AC 29 ms
10,436 KB
testcase_31 AC 31 ms
10,436 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#line 1 "/home/siro53/kyo-pro/compro_library/template/template.cpp"
#include <bits/stdc++.h>
using namespace std;
template <class T> inline bool chmax(T &a, T b) {
    if(a < b) {
        a = b;
        return 1;
    }
    return 0;
}
template <class T> inline bool chmin(T &a, T b) {
    if(a > b) {
        a = b;
        return 1;
    }
    return 0;
}
#define DEBUG
#ifdef DEBUG
template <class T, class U>
ostream &operator<<(ostream &os, const pair<T, U> &p) {
    os << '(' << p.first << ',' << p.second << ')';
    return os;
}
template <class T> ostream &operator<<(ostream &os, const vector<T> &v) {
    os << '{';
    for(int i = 0; i < (int)v.size(); i++) {
        if(i) { os << ','; }
        os << v[i];
    }
    os << '}';
    return os;
}
void debugg() { cerr << endl; }
template <class T, class... Args>
void debugg(const T &x, const Args &... args) {
    cerr << " " << x;
    debugg(args...);
}
#define debug(...)                                                             \
    cerr << __LINE__ << " [" << #__VA_ARGS__ << "]: ", debugg(__VA_ARGS__)
#define dump(x) cerr << __LINE__ << " " << #x << " = " << (x) << endl
#else
#define debug(...) (void(0))
#define dump(x) (void(0))
#endif

struct Setup {
    Setup() {
        cin.tie(0);
        ios::sync_with_stdio(false);
        cout << fixed << setprecision(15);
    }
} __Setup;

using ll = long long;
#define ALL(v) (v).begin(), (v).end()
#define RALL(v) (v).rbegin(), (v).rend()
#define FOR(i, a, b) for(int i = (a); i < int(b); i++)
#define REP(i, n) FOR(i, 0, n)
const int INF = 1 << 30;
const ll LLINF = 1LL << 60;
constexpr int MOD = 1000000007;
const int dx[4] = {1, 0, -1, 0};
const int dy[4] = {0, 1, 0, -1};

//-------------------------------------
#line 1 "/home/siro53/kyo-pro/compro_library/graph/tree_diameter.hpp"
vector<int> get_diameter_path(const vector<vector<int>> &G) {
    int n = (int)G.size();
    vector<int> depth(n), par(n);

    auto dfs = [&](auto &&dfs, int v, int p, int d) -> void {
        depth[v] = d;
        par[v] = p;
        for(const int &u : G[v]) {
            if(u != p) {
                dfs(dfs, u, v, d + 1);
            }
        }
    };

    dfs(dfs, 0, -1, 0);
    int from = max_element(ALL(depth)) - depth.begin();
    dfs(dfs, from, -1, 0);
    int to = max_element(ALL(depth)) - depth.begin();

    vector<int> res;
    int now = to;
    while(now != from) {
        res.emplace_back(now);
        now = par[now];
    }
    res.emplace_back(now);

    return res;
}
#line 3 "c.cpp"

int N;
vector<vector<int>> G;
vector<int> leafs, depth;

void dfs(int u, int p, int d) {
    bool is_leaf = true;
    depth[u] = d;
    for(int v : G[u]) {
        if(v == p) continue;
        is_leaf = false;
        dfs(v, u, d+1);
    }
    if(is_leaf) leafs.push_back(u);
}

int main() {
    cin >> N; G.resize(N);
    REP(i, N-1) {
        int a, b; cin >> a >> b; a--; b--;
        G[a].push_back(b); G[b].push_back(a);
    }

    auto dia = get_diameter_path(G);
    int len = dia.size();
    int center = dia[len / 2];

    depth.resize(N);
    dfs(center, -1, 0);

    set<int> se;
    for(int leaf : leafs) se.emplace(depth[leaf]);

    if(se.size() == 1) {
        cout << "Yes\n";
    } else{
        cout << "No\n";
    }
}
0