結果
問題 | No.386 貪欲な領主 |
ユーザー |
![]() |
提出日時 | 2016-08-31 03:55:14 |
言語 | C++11 (gcc 13.3.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 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 16 |
ソースコード
#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;}