結果

問題 No.806 木を道に
ユーザー ugis_progugis_prog
提出日時 2019-03-22 23:41:48
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 1,542 bytes
コンパイル時間 877 ms
コンパイル使用メモリ 86,896 KB
実行使用メモリ 11,084 KB
最終ジャッジ日時 2023-10-19 10:56:47
合計ジャッジ時間 3,023 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 2 ms
4,348 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 AC 2 ms
4,348 KB
testcase_05 WA -
testcase_06 WA -
testcase_07 AC 2 ms
4,348 KB
testcase_08 AC 2 ms
4,348 KB
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 AC 46 ms
8,748 KB
testcase_25 AC 68 ms
11,084 KB
testcase_26 AC 22 ms
5,480 KB
testcase_27 AC 70 ms
10,188 KB
testcase_28 AC 3 ms
4,348 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<unordered_set>
using namespace std;

int main(){
    int N;
    cin >> N;
    vector<vector<int>> G(N);
    vector<bool> visited(N,false);
    vector<int> back(N,-1);

    int start = -1;
    for(int i=0;i<N-1;i++){
        int a,b;
        cin >> a >> b;
        a--, b--;
        G[a].push_back(b);
        G[b].push_back(a);
        if(G[a].size() == 1) start = a;
        if(G[b].size() == 1) start = b;
    }

    queue<pair<int,int>> Q;
    Q.push({start,0});
    visited[start] = true;
    int end = 0;

    int dia = 0;
    while(!Q.empty()){
        pair<int,int> p = Q.front();
        Q.pop();

        for(int to : G[p.first]){
            if(visited[to]) continue;
            visited[to] = true;
            if(dia <= p.second+1){
                back[to] = p.first;
                end = to;
                dia = p.second+1;
            }
            Q.push({to,p.second+1});
        }
    }

    unordered_set<int> st;

    while(end != start){
        st.insert(end);
        end = back[end];
    }
    st.insert(start);
    
    int ans = 0;
    for(int i=0;i<N;i++){
        if(st.count(i)) continue;
        if(G[i].size() > 0){
            for(int to : G[i]){
                if(st.count(to)){
                    if(G[i].size() == 1) ans++;
                    else ans += G[i].size() - 1;
                    break;
                }
                else ans += G[i].size() - 1;
            }
        }
    }

    cout << ans << endl;

}
0