結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2021-03-06 00:38:59 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 51 ms / 2,000 ms |
コード長 | 832 bytes |
コンパイル時間 | 1,785 ms |
コンパイル使用メモリ | 196,152 KB |
最終ジャッジ日時 | 2025-01-19 12:10:46 |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
コンパイルメッセージ
main.cpp: In function ‘int main()’: main.cpp:26:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 26 | scanf("%ld", &n); | ~~~~~^~~~~~~~~~~ main.cpp:29:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 29 | scanf("%d %d", &a, &b); | ~~~~~^~~~~~~~~~~~~~~~~
ソースコード
#include<bits/stdc++.h> #define REP(i,b,e) for(int i=b;i<e;i++) using ll = int_fast64_t; std::vector<int> links[100000], visit; int in[100000], out[100000], from[100000]; ll cnt[100001]; bool used[100000]; void eular_tour(int v=0){ if(used[v]) return; used[v] = true; in[v] = visit.size(); visit.push_back(v); for(auto to: links[v]){ if(!used[to]){ from[to] = v; eular_tour(to); } } out[v] = visit.size(); } int main(){ ll n; scanf("%ld", &n); REP(i, 0, n-1){ int a, b; scanf("%d %d", &a, &b); links[a-1].push_back(b-1); links[b-1].push_back(a-1); } eular_tour(); REP(i, 0, n) cnt[i] = out[i] - in[i]; ll ans = 0; REP(i, 0, n){ ans += n; for(auto x: links[i]){ if(from[i]!=x) ans += cnt[x] * (n - cnt[x]); else ans += cnt[i] * (n - cnt[i]); } } printf("%ld\n", ans); return 0; }