結果

問題 No.1817 Reversed Edges
ユーザー se1ka2se1ka2
提出日時 2022-01-21 21:50:57
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 245 ms / 2,000 ms
コード長 3,360 bytes
コンパイル時間 864 ms
コンパイル使用メモリ 81,828 KB
実行使用メモリ 21,640 KB
最終ジャッジ日時 2023-08-17 05:41:48
合計ジャッジ時間 6,411 ms
ジャッジサーバーID
(参考情報)
judge12 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 2 ms
4,380 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 2 ms
4,380 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 210 ms
12,540 KB
testcase_08 AC 101 ms
7,644 KB
testcase_09 AC 200 ms
12,076 KB
testcase_10 AC 112 ms
8,264 KB
testcase_11 AC 147 ms
9,648 KB
testcase_12 AC 245 ms
13,924 KB
testcase_13 AC 243 ms
13,856 KB
testcase_14 AC 241 ms
13,840 KB
testcase_15 AC 242 ms
13,920 KB
testcase_16 AC 239 ms
13,916 KB
testcase_17 AC 237 ms
13,748 KB
testcase_18 AC 244 ms
13,932 KB
testcase_19 AC 243 ms
13,764 KB
testcase_20 AC 238 ms
13,816 KB
testcase_21 AC 240 ms
13,860 KB
testcase_22 AC 186 ms
13,196 KB
testcase_23 AC 185 ms
12,912 KB
testcase_24 AC 213 ms
21,640 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <queue>
using namespace std;

struct Tree
{
    int n;
    int r;
    int l;
    std::vector<std::vector<int>> prit;
    std::vector<std::vector<int>> t;
    std::vector<int> par;
    std::vector<int> dpt;
    std::vector<int> tour;
    std::vector<std::vector<int>> appear_on_tour;
    std::vector<std::vector<int>> anc;
    
    Tree(){}
    
    Tree(int n) : n(n){
        prit.resize(n);
        t.resize(n);
        par.resize(n);
        dpt.resize(n);
    }
    
    void add_edge(int u, int v){
        prit[u].push_back(v);
        prit[v].push_back(u);
    }
    
    void build(int r_){
        r = r_;
        for(int i = 0; i < n; i++) dpt[i] = -1;
        std::queue<int> que;
        par[r] = -1;
        dpt[r] = 0;
        que.push(r);
        while(que.size()){
            int u = que.front();
            que.pop();
            for(int v : prit[u]){
                if(dpt[v] == -1){
                    t[u].push_back(v);
                    par[v] = u;
                    dpt[v] = dpt[u] + 1;
                    que.push(v);
                }
            }
        }
    }
    
    void rebuild(int r_){
        for(int u = 0; u < n; u++) t[u].clear();
        build(r_);
    }
    
/*
    std::vector<int> toposort(){
        std::vector<std::pair<int, int>> p(n);
        for(int i = 0; i < n; i++) p[i] = std::pair<int, int>(dpt[i], i);
        sort(p.begin(), p.end(), std::greater<std::pair<int, int>>());
        std::vector<int> res(n);
        for(int i = 0; i < n; i++) res[i] = p[i].second;
        return res;
    }
*/
    
    void dfs(int u, int &k){
        appear_on_tour[u].push_back(k);
        tour[k++] = u;
        for(int v : t[u]){
            dfs(v, k);
            appear_on_tour[u].push_back(k);
            tour[k++] = u;
        }
    }
    
    void build_euler_tour(){
        tour.resize(n * 2 - 1);
        appear_on_tour.resize(n);
        int k = 0;
        dfs(r, k);
    }
    
    void build_lca(){
        for(l = 1; l <= n; l++){
            if((1 << (l - 1)) >= n) break;
        }
        anc.resize(l);
        for(int j = 0; j < l; j++) anc[j].resize(n);
        for(int i = 0; i < n; i++) anc[0][i] = par[i];
        for(int j = 1; j < l; j++){
            for(int i = 0; i < n; i++){
                if(anc[j - 1][i] == -1) anc[j][i] = -1;
                else anc[j][i] = anc[j - 1][anc[j - 1][i]];
            }
        }
    }
    
    int lca(int u, int v){
        if(dpt[u] < dpt[v]) std::swap(u, v);
        int dpt_diff = dpt[u] - dpt[v];
        for(int j = l - 1; j >= 0; j--){
            if((dpt_diff >> j) & 1) u = anc[j][u];
        }
        if(u == v) return u;
        for(int j = l - 1; j >= 0; j--){
            if(anc[j][u] != anc[j][v]){
                u = anc[j][u];
                v = anc[j][v];
            }
        }
        return anc[0][u];
    }
};

void dfs(int u, int *ans, Tree &t){
	for(int v : t.t[u]){
		if(v < u) ans[v] = ans[u] - 1;
		else ans[v] = ans[u] + 1;
		dfs(v, ans, t);
	}
}

int main()
{
	int n;
	cin >> n;
	Tree t(n);
	for(int i = 0; i < n - 1; i++){
		int a, b;
		cin >> a >> b;
		a--; b--;
		t.add_edge(a, b);
	}
	t.build(0);
	int ans[100005];
	ans[0] = 0;
	for(int u = 0; u < n; u++){
		for(int v : t.t[u]){
			if(v < u) ans[0]++;
		}
	}
	dfs(0, ans, t);
	for(int u = 0; u < n; u++) cout << ans[u] << endl;
}
0