結果
問題 | No.1098 LCAs |
ユーザー |
![]() |
提出日時 | 2020-06-26 23:10:50 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 209 ms / 2,000 ms |
コード長 | 2,387 bytes |
コンパイル時間 | 1,016 ms |
コンパイル使用メモリ | 107,584 KB |
実行使用メモリ | 40,448 KB |
最終ジャッジ日時 | 2024-07-04 23:39:37 |
合計ジャッジ時間 | 4,551 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
#include <iostream> // cout, endl, cin#include <string> // string, to_string, stoi#include <vector> // vector#include <algorithm> // min, max, swap, sort, reverse, lower_bound, upper_bound#include <utility> // pair, make_pair#include <tuple> // tuple, make_tuple#include <cstdint> // int64_t, int*_t#include <cstdio> // printf#include <map> // map#include <queue> // queue, priority_queue#include <set> // set#include <stack> // stack#include <deque> // deque#include <unordered_map> // unordered_map#include <unordered_set> // unordered_set#include <bitset> // bitset#include <cctype> // isupper, islower, isdigit, toupper, tolower#include <iomanip> // setprecision#include <complex> // complex#include <math.h>#include <climits>#include <assert.h>#define rep(i,n) for(int i=0; i<(n); i++)using namespace std;using ll = long long;using P = pair<int,int>;constexpr ll INF = 1e18;constexpr int inf = 1e9;constexpr ll mod = 1000000007;constexpr ll mod2 = 998244353;const double PI = 3.1415926535897932384626433832795028841971;const int dx[8] = {1, 0, -1, 0,1,1,-1,-1};const int dy[8] = {0, 1, 0, -1,1,-1,1,-1};template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }// ---------------------------------------------------------------------------int N;vector<vector<int>> G;vector<vector<int>> ch;vector<int> sum;void dfs(int v,int p=-1){sum[v] = 1;for(int nv: G[v]){if(nv == p) continue;dfs(nv,v);sum[v] += sum[nv];ch[v].emplace_back(sum[nv]);}}int main(){ios::sync_with_stdio(false);cin.tie(nullptr);cin >> N;G.resize(N);for(int i=0; i<N-1; i++){int a,b;cin >> a >> b;a--;b--;G[a].emplace_back(b);G[b].emplace_back(a);}ch.resize(N);sum = vector<int>(N,0);dfs(0);// for(int i=0; i<N; i++){// cout << ch[i].size() << "\n";// for(int j=0; j<ch[i].size(); j++){// cout << ch[i][j] << " \n"[j+1==ch[i].size()];// }// }for(int i=0; i<N; i++){ll ans = sum[i];for(int j=0; j<ch[i].size(); j++){ans += (ll)ch[i][j] * (sum[i]-ch[i][j]);}cout << ans << "\n";}return 0;}