提出 #34047310

---

ソースコード 拡げる

#include <bits/stdc++.h>
#define pb push_back
#define ll long long
#define all(x) begin(x),end(x)
using namespace std;

const int n0=3e5+123;
const ll inf=1e18;
int n,d[n0];
vector<array<int,2>> g[n0];
vector<ll> dp[n0];
ll mx0[n0],mx1[n0];

void dfs(int v,int p) {
	ll def=0;
	vector<ll> vec;
	for(auto i:g[v]) {
		int to=i[0],w=i[1];
		if(to!=p) {
			dfs(to,v);
			def+=max(0ll,mx0[to]);
			vec.pb(max(0ll,mx1[to]+w-mx0[to]));
		}
	}
	sort(all(vec));reverse(all(vec));
	for(int i=0; i<=d[v]; i++) {
		if(i && i<=(int)vec.size()) {
			def+=vec[i-1];
		}
		dp[v][d[v]-i]=def;
	}
	for(int i=0; i<=d[v]; i++) {
		mx0[v]=max(mx0[v],dp[v][i]);
	}
	for(int i=1; i<=d[v]; i++) {
		mx1[v]=max(mx1[v],dp[v][i]);
	}
}

int main() {
	ios_base::sync_with_stdio(0);cin.tie(0);
	cin >> n;
	for(int i=1; i<=n; i++) {
		mx0[i]=mx1[i]=-inf;
		cin >> d[i];
		dp[i]=vector<ll>(d[i]+1,0);
	}
	for(int i=1; i<n; i++) {
		int x,y,z;cin >> x >> y >> z;g[x].pb({y,z});g[y].pb({x,z});
	}
	dfs(1,-1);
	cout << mx0[1];
}

提出情報

ジャッジ結果