結果
問題 | No.1098 LCAs |
ユーザー |
![]() |
提出日時 | 2020-08-05 05:14:04 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 694 ms / 2,000 ms |
コード長 | 1,774 bytes |
コンパイル時間 | 3,251 ms |
コンパイル使用メモリ | 199,432 KB |
最終ジャッジ日時 | 2025-01-12 14:40:11 |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
コンパイルメッセージ
main.cpp: In function ‘int main()’: main.cpp:46:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 46 | scanf("%d%d", &a, &b); | ~~~~~^~~~~~~~~~~~~~~~
ソースコード
#include <bits/stdc++.h>#define rep(i,n) for(int i = 0; i < (n); ++i)#define drep(i,n) for(int i = (n)-1; i >= 0; --i)#define srep(i,s,t) for (int i = s; i < t; ++i)using namespace std;typedef long long int ll;typedef pair<int,int> P;#define yn {puts("Yes");}else{puts("No");}#define MAX_N 200005vector<int> pre_G[MAX_N];vector<int> G[MAX_N];int par_[MAX_N];int flag_netsukigi[MAX_N];void make_netsukigi(int r){rep(i,MAX_N){G[i].clear();par_[i] = 0;flag_netsukigi[i] = 0;}queue<int> que;que.push(r);flag_netsukigi[r] = 1;par_[r] = -1;while(que.size() > 0){int x = que.front();que.pop();flag_netsukigi[x] = 1;rep(i,pre_G[x].size()){if(flag_netsukigi[pre_G[x][i]] == 0){par_[pre_G[x][i]] = x;G[x].push_back(pre_G[x][i]);que.push(pre_G[x][i]);}}}}int main() {int n;cin >> n;rep(i,n-1){int a, b;scanf("%d%d", &a, &b);a--; b--; // 0-indexにするため.pre_G[a].push_back(b);pre_G[b].push_back(a);}make_netsukigi(0);ll ans[n] = {};int son[n];queue<int> que;ll dp[n] = {};rep(i,n){son[i] = G[i].size();if(son[i] == 0){que.push(i);}}while(que.size() > 0){int x = que.front();que.pop();rep(i,G[x].size()){ans[x] += (dp[G[x][i]]+1) * (dp[x]-(dp[G[x][i]]+1));}ans[x] += dp[x]*2+1;int y = par_[x];if(y==-1)break;dp[y] += dp[x] + 1;son[y]--;if(son[y]==0)que.push(y);}rep(i,n) cout << ans[i] << endl;return 0;}