結果

問題 No.386 貪欲な領主
ユーザー どららどらら
提出日時 2016-08-07 12:52:26
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 326 ms / 2,000 ms
コード長 1,894 bytes
コンパイル時間 855 ms
コンパイル使用メモリ 83,128 KB
実行使用メモリ 41,188 KB
最終ジャッジ日時 2024-11-07 07:14:42
合計ジャッジ時間 3,999 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
6,016 KB
testcase_01 AC 4 ms
6,016 KB
testcase_02 AC 4 ms
6,016 KB
testcase_03 AC 4 ms
6,016 KB
testcase_04 AC 231 ms
41,188 KB
testcase_05 AC 288 ms
35,152 KB
testcase_06 AC 326 ms
35,072 KB
testcase_07 AC 6 ms
6,016 KB
testcase_08 AC 32 ms
8,960 KB
testcase_09 AC 7 ms
6,144 KB
testcase_10 AC 4 ms
6,016 KB
testcase_11 AC 3 ms
6,016 KB
testcase_12 AC 5 ms
6,016 KB
testcase_13 AC 9 ms
6,912 KB
testcase_14 AC 316 ms
35,200 KB
testcase_15 AC 188 ms
41,180 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cctype>
#include <cmath>
#include <iostream>
#include <queue>
#include <list>
#include <stack>
#include <map>
#include <numeric>
#include <set>
#include <sstream>
#include <string>
#include <vector>
using namespace std;
#define REP(i,a,n) for(int i=(a); i<(int)(n); i++)
#define rep(i,n) REP(i,0,n)
#define FOR(it,c) for(__typeof((c).begin()) it=(c).begin(); it!=(c).end(); ++it)
#define ALLOF(c) (c).begin(), (c).end()
typedef long long ll;

int N;
vector<int> G[100005];
int parent[64][100005];
int depth[100005];

int dp[100005];
int weight[100005];

void dfs(int v, int p, int d){
  parent[0][v] = p;
  depth[v] = d;
  dp[v] = (p!=-1?dp[p]:0) + weight[v];
  for(int i=0; i<G[v].size(); i++){
    if(G[v][i] != p) dfs(G[v][i], v, d+1);
  }
}
void init(int root){
  dfs(root, -1, 0);
  for(int k=0; k+1<64; k++){
    for(int v=0; v<N+1; v++){
      if(parent[k][v] < 0) parent[k+1][v] = -1;
      else parent[k+1][v] = parent[k][parent[k][v]];
    }
  }
}
int lca(int u, int v){
  if(depth[u] > depth[v]) swap(u, v);
  for(int k=0; k<64; k++){
    if((depth[v] - depth[u]) >> k & 1){
      v = parent[k][v];
    }
  }
  if(u==v) return u;
  for(int k=64-1; k>=0; k--){
    if(parent[k][u] != parent[k][v]){
      u = parent[k][u];
      v = parent[k][v];
    }
  }
  return parent[0][u];
}


int main(){
  ios::sync_with_stdio(false);
  cin >> N;
  rep(i,N-1){
    int a, b;
    cin >> a >> b;
    G[a].push_back(b);
    G[b].push_back(a);
  }
  rep(i,N){
    cin >> weight[i];
  }
  dp[0] = weight[0];
  init(0);
  
  int M;
  cin >> M;
  
  ll ret = 0;
  rep(i,M){
    int a, b, c;
    cin >> a >> b >> c;
    int l = lca(a, b);
    //cout << " " << dp[a] << " " << dp[b] << " " << dp[l] << " " << weight[l] << endl;
    ret += (dp[a] + dp[b] - 2LL*dp[l] + weight[l]) * (ll)c;
  }
  cout << ret << endl;
  
  return 0;
}
0