結果
問題 | No.1103 Directed Length Sum |
ユーザー | hs0319boy |
提出日時 | 2020-07-03 23:14:23 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 1,507 ms / 3,000 ms |
コード長 | 1,895 bytes |
コンパイル時間 | 1,906 ms |
コンパイル使用メモリ | 182,896 KB |
実行使用メモリ | 77,676 KB |
最終ジャッジ日時 | 2024-09-17 04:53:18 |
合計ジャッジ時間 | 15,166 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,812 KB |
testcase_01 | AC | 2 ms
6,940 KB |
testcase_02 | AC | 737 ms
77,676 KB |
testcase_03 | AC | 735 ms
64,764 KB |
testcase_04 | AC | 803 ms
41,592 KB |
testcase_05 | AC | 1,507 ms
66,112 KB |
testcase_06 | AC | 485 ms
27,292 KB |
testcase_07 | AC | 93 ms
9,088 KB |
testcase_08 | AC | 148 ms
12,480 KB |
testcase_09 | AC | 57 ms
6,944 KB |
testcase_10 | AC | 225 ms
15,480 KB |
testcase_11 | AC | 876 ms
44,644 KB |
testcase_12 | AC | 498 ms
27,580 KB |
testcase_13 | AC | 228 ms
15,784 KB |
testcase_14 | AC | 42 ms
6,944 KB |
testcase_15 | AC | 368 ms
22,656 KB |
testcase_16 | AC | 1,005 ms
49,148 KB |
testcase_17 | AC | 1,069 ms
51,860 KB |
testcase_18 | AC | 213 ms
15,380 KB |
testcase_19 | AC | 931 ms
46,860 KB |
testcase_20 | AC | 72 ms
7,936 KB |
testcase_21 | AC | 138 ms
11,136 KB |
testcase_22 | AC | 745 ms
36,164 KB |
testcase_23 | AC | 405 ms
23,872 KB |
コンパイルメッセージ
main.cpp: In function 'int main()': main.cpp:40:22: warning: 'root' may be used uninitialized [-Wmaybe-uninitialized] 40 | vin d(n+1);d[root]=1; | ^ main.cpp:37:9: note: 'root' was declared here 37 | int root; | ^~~~
ソースコード
#include <bits/stdc++.h> using namespace std; using ll=long long; using vin=vector<int>; using vll=vector<long long>; using vvin=vector<vector<int>>; using vvll=vector<vector<long long>>; using vstr=vector<string>; using vvstr=vector<vector<string>>; using vch=vector<char>; using vvch=vector<vector<char>>; using vbo=vector<bool>; using vvbo=vector<vector<bool>>; using vpii=vector<pair<int,int>>; using pqsin=priority_queue<int,vector<int>,greater<int>>; #define mp make_pair #define rep(i,n) for(int i=0;i<(int)(n);i++) #define rep2(i,s,n) for(int i=(s);i<(int)(n);i++) #define all(v) v.begin(),v.end() #define decp(n) cout<<fixed<<setprecision((int)n) const ll inf=1e9+7; const ll INF=1e18; int main(){ int n;cin>>n; vin a(n-1),b(n-1); vin par(n+1,-1); vvin baby(n+1); vbo r(n+1,false); rep(i,n-1){ cin>>a[i]>>b[i]; baby[a[i]].push_back(b[i]); par[b[i]]=a[i]; r[b[i]]=true; } int root; rep(i,n)if(!r[i+1])root=i+1; queue<int> qq;qq.push(root); vin d(n+1);d[root]=1; while(qq.size()){ auto p=qq.front();qq.pop(); if(baby[p].size()==0)continue; for(auto tmp:baby[p]){ d[tmp]=d[p]+1; qq.push(tmp); } } vin child(n+1,1);//the number of children vbo arrived(n+1,false); priority_queue<pair<int,int>> q; rep(i,n)if(baby[i+1].size()==0)q.push(mp(d[i+1],i+1)); while(q.size()){ auto p=q.top().second;q.pop(); //cout<<p<<" "; if(arrived[p]||par[p]==-1)continue; child[par[p]]+=child[p]; q.push(mp(d[par[p]],par[p])); arrived[p]=true; } //cout<<endl; ll ans=(ll)0; rep(i,n-1){ ans+=((ll)d[a[i]]*(ll)child[b[i]])%inf; ans%=inf; } cout<<ans<<endl; /*rep(i,n)cout<<par[i+1]<<" "; cout<<endl; rep(i,n)cout<<child[i+1]<<" "; cout<<endl;*/ }