結果
問題 | No.386 貪欲な領主 |
ユーザー | uenoku |
提出日時 | 2016-08-31 03:55:14 |
言語 | C++11 (gcc 11.4.0) |
結果 |
AC
|
実行時間 | 358 ms / 2,000 ms |
コード長 | 2,169 bytes |
コンパイル時間 | 781 ms |
コンパイル使用メモリ | 90,664 KB |
実行使用メモリ | 25,076 KB |
最終ジャッジ日時 | 2024-11-14 12:43:35 |
合計ジャッジ時間 | 3,350 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 20 ms
15,232 KB |
testcase_01 | AC | 16 ms
15,232 KB |
testcase_02 | AC | 19 ms
15,300 KB |
testcase_03 | AC | 19 ms
15,296 KB |
testcase_04 | AC | 358 ms
25,076 KB |
testcase_05 | AC | 263 ms
19,120 KB |
testcase_06 | AC | 263 ms
18,860 KB |
testcase_07 | AC | 20 ms
15,300 KB |
testcase_08 | AC | 51 ms
15,432 KB |
testcase_09 | AC | 23 ms
15,044 KB |
testcase_10 | AC | 16 ms
15,300 KB |
testcase_11 | AC | 19 ms
15,304 KB |
testcase_12 | AC | 20 ms
15,168 KB |
testcase_13 | AC | 25 ms
15,428 KB |
testcase_14 | AC | 267 ms
18,948 KB |
testcase_15 | AC | 291 ms
24,920 KB |
ソースコード
#include <iostream> #include <string> #include <vector> #include <algorithm> #include <cstdio> #include <queue> #include "math.h" #include <complex> #include <iomanip> #include <map> #define ifor(i,a,b) for (int i=(a);i<(b);i++) #define rfor(i,a,b) for (int i=(b)-1;i>=(a);i--) #define rep(i,n) for (int i=0;i<(n);i++) #define rrep(i,n) for (int i=(n)-1;i>=0;i--) using namespace std; typedef long double ld; typedef long long int lli; typedef complex <double> P; const double eps = 1e-11; int vex[4]={1,0,-1,0}; int vey[4]={0,1,0,-1}; typedef vector<double> Vec; typedef vector<int> vec; typedef vector<Vec> MAT; typedef vector<vec> mat; lli MOD=1000000007; vector<int> E[100005]; int par[100005][20]; int depth[100005]; void dfs(int v,int p,int d) { par[v][0]=p; depth[v]=d; for(int i=0;i<E[v].size();i++) { if(E[v][i]==p)continue; dfs(E[v][i],v,d+1); } } void search_depth(int root){ dfs(root,-1,0); } void fill_table(){ rep(i,19){ rep(j,100004){ if(par[j][i] == -1)par[j][i+1] =-1; else par[j][i+1]= par[par[j][i]][i]; } } } int lca(int u,int v) { if(depth[u]>depth[v])swap(u,v); for(int i=19;i>=0;i--) { if(((depth[v]-depth[u])>>i)&1)v=par[v][i]; } if(u==v)return u; for(int i=19;i>=0;i--) { if(par[u][i]!=par[v][i]) { u = par[u][i]; v = par[v][i]; } } return par[u][0]; } int main(){ int N; cin >> N; int a,b,c; par[0][0]=-1; rep(i,N-1){ cin>>a >>b; E[a].push_back(b); E[b].push_back(a); } queue<int> que; que.push(0); lli cost [100005],x[100005]; rep(i,N)cin>>x[i]; cost[0]= x[0]; bool used[100005]; while(!que.empty()){ int t = que.front();que.pop(); used[t]=true; rep(i,E[t].size()){ int to = E[t][i]; if(used[to])continue; cost[to] = cost[t]+x[to]; que.push(to); } } search_depth(0); //rep(i,10)printf("(%d,%d)",i,depth[i]); fill_table(); int M; cin >> M; lli ans =0; rep(i,M){ cin>>a>>b>>c; //cout<<lca(a,b)<<endl; int p = lca(a,b); //printf("%d %d %d\n",cost[a],cost[b],cost[p]); ans+= (cost[a]+cost[b]-2*cost[p]+x[p])*c; } cout<<ans<<endl; }