結果

問題 No.2427 Tree Distance Two
コンテスト
ユーザー chro_96
提出日時 2023-08-18 22:04:11
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 167 ms / 2,000 ms
コード長 1,459 bytes
コンパイル時間 367 ms
コンパイル使用メモリ 29,952 KB
実行使用メモリ 46,208 KB
最終ジャッジ日時 2024-11-28 07:14:56
合計ジャッジ時間 5,402 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>

typedef struct list {
  int v;
  struct list *n;
} list;

void func1 (int v, list **e, int tmp_d, int *d, int *cnt, int *visited, int *ans) {
  list *l = e[v];
  
  d[v] = tmp_d;
  visited[v] = 1;
  while (l != NULL) {
    if (visited[l->v] <= 0) {
      func1(l->v, e, tmp_d+1, d, cnt, visited, ans);
      cnt[v]++;
      ans[v] += cnt[l->v];
    }
    l = l->n;
  }
  
  return;
}

void func2 (int v, list **e, int *cnt, int *visited, int *ans) {
  list *l = e[v];
  
  visited[v] = 2;
  while (l != NULL) {
    if (visited[l->v] <= 1) {
      func2(l->v, e, cnt, visited, ans);
      ans[l->v] += cnt[v]-1;
    }
    l = l->n;
  }
  
  return;
}

int main () {
 int n = 0;
 int u = 0;
 int v = 0;
 
 int res = 0;
 
 int ans[300000] = {};
 
 int d[300000] = {};
 int dcnt[300002] = {};
 
 list pool[599998] = {};
 int used = 0;
 
 list *e[300000] = {};
 
 int visited[300000] = {};
 int cnt[300000] = {};
 
 res = scanf("%d", &n);
 for (int i = 0; i < n-1; i++) {
   res = scanf("%d", &u);
   res = scanf("%d", &v);
   u--;
   v--;
   pool[used].v = v;
   pool[used].n = e[u];
   e[u] = pool+used;
   used++;
   pool[used].v = u;
   pool[used].n = e[v];
   e[v] = pool+used;
   used++;
 }
 
 func1(0, e, 0, d, cnt, visited, ans);
 func2(0, e, cnt, visited, ans);
 
 for (int i = 0; i < n; i++) {
   dcnt[d[i]]++;
 }
 
 for (int i = 0; i < n; i++) {
   if (d[i] > 1) {
     ans[i] += 1;
   }
   printf("%d\n", ans[i]);
 }
 
 return 0;
}
0