結果
問題 | No.1221 木 *= 3 |
ユーザー | Shibuyap |
提出日時 | 2020-09-04 22:03:27 |
言語 | C++17 (gcc 13.2.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 150 ms / 2,000 ms |
コード長 | 1,963 bytes |
コンパイル時間 | 2,048 ms |
コンパイル使用メモリ | 206,688 KB |
実行使用メモリ | 25,280 KB |
最終ジャッジ日時 | 2023-08-17 15:40:16 |
合計ジャッジ時間 | 5,457 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 7 ms
15,316 KB |
testcase_01 | AC | 6 ms
15,420 KB |
testcase_02 | AC | 6 ms
15,304 KB |
testcase_03 | AC | 6 ms
15,380 KB |
testcase_04 | AC | 5 ms
15,456 KB |
testcase_05 | AC | 6 ms
15,352 KB |
testcase_06 | AC | 6 ms
15,400 KB |
testcase_07 | AC | 143 ms
25,280 KB |
testcase_08 | AC | 139 ms
25,132 KB |
testcase_09 | AC | 148 ms
25,104 KB |
testcase_10 | AC | 150 ms
25,076 KB |
testcase_11 | AC | 141 ms
25,188 KB |
testcase_12 | AC | 125 ms
22,784 KB |
testcase_13 | AC | 119 ms
22,780 KB |
testcase_14 | AC | 124 ms
22,972 KB |
testcase_15 | AC | 123 ms
22,940 KB |
testcase_16 | AC | 123 ms
22,936 KB |
testcase_17 | AC | 146 ms
24,084 KB |
testcase_18 | AC | 142 ms
23,880 KB |
testcase_19 | AC | 141 ms
24,100 KB |
testcase_20 | AC | 138 ms
24,012 KB |
testcase_21 | AC | 141 ms
23,884 KB |
ソースコード
#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 200005 vector<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]); } } } } ll dp[MAX_N][2]; ll INF = 1001001001001001001; int main() { int n; cin >> n; ll a[n], b[n]; rep(i,n) cin >> a[i]; rep(i,n) cin >> b[i]; 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); rep(i,n)rep(j,2)dp[i][j] = -INF; queue<int> que; int son[n] = {}; rep(i,n){ son[i] = G[i].size(); if(son[i] == 0){ que.push(i); } } while(que.size()){ int x = que.front(); que.pop(); dp[x][0] = a[x]; dp[x][1] = 0; rep(i,G[x].size()){ int y = G[x][i]; dp[x][0] += max(dp[y][0], dp[y][1]); dp[x][1] += max(dp[y][0], dp[y][1] + b[x] + b[y]); } if(x == 0)break; son[par_[x]]--; if(son[par_[x]] == 0) que.push(par_[x]); } cout << max(dp[0][0], dp[0][1]) << endl; return 0; }