Submission #670212

#TimeUsernameProblemLanguageResultExecution timeMemory
670212radalBeads and wires (APIO14_beads)C++17
100 / 100
172 ms25408 KiB
#include <bits/stdc++.h>
#pragma GCC target("sse,sse2,avx2")
#pragma GCC optimize("unroll-loops,O2")
#define rep(i,l,r) for (int i = l; i < r; i++)
#define repr(i,r,l) for (int i = r; i >= l; i--)
#define X first
#define Y second
#define all(x) (x).begin() , (x).end()
#define pb push_back
#define endl '\n'
#define debug(x) cerr << #x << " : " << x << endl;
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
constexpr int N = 2e5+10,mod = 1e9+7,maxm = 210;
constexpr ll inf = 1e9+10;
inline int mkay(int a,int b){
    if (a+b >= mod) return a+b-mod;
   // if (a+b < 0) return a+b+mod;
    return a+b;
}
 
inline int poww(int a,int k){
    if (k < 0) return 0;
    int z = 1;
    while (k){
        if (k&1) z = 1ll*z*a%mod;
        a = 1ll*a*a%mod;
        k >>= 1;
    } 
    return z; 
}
 
vector<pll> adj[N];
int dp[N][2][2]; // good e
 
void dfs(int v,int p){
	if ((int) adj[v].size() == 1 && p != 0){
		dp[v][0][1] = dp[v][1][1] = -2*inf;
		dp[v][0][0] = dp[v][1][0] = 0;
		return;
	}
	int s = 0,mx = -2*inf;
	int mx2 = -2*inf;
	vector<int> ve;
	for (auto [u,w] : adj[v]){
		if (u == p) continue;
		dfs(u,v);
		int val = max(dp[u][1][0],dp[u][1][1]+w);
		s += val;
		mx = max(mx,dp[u][1][0]-val+w);

		int val2 = max(dp[u][0][0],dp[u][0][1]+w);
		mx2 = max(mx2,val2-val);
		ve.pb(dp[u][0][0]-val+w);
	}
	sort(all(ve));
	dp[v][1][0] = s;
	dp[v][1][1] = s + mx;
	dp[v][0][0] = s;
	dp[v][0][0] = max(dp[v][0][0],s+mx2);
	dp[v][0][1] = s+ve.back();
	int sz = ve.size();
	if (sz == 1) return;
	for (auto [u,w] : adj[v]){
		if (u == p) continue;
        int val = max(dp[u][1][0],dp[u][1][1]+w);
		if (ve.back() != dp[u][0][0]-val+w){
			dp[v][0][0] = max(dp[v][0][0],s-val+ve.back()+w+dp[u][1][0]);
		}
		else{
            dp[v][0][0] = max(dp[v][0][0],s-val+ve[sz-2]+w+dp[u][1][0]);
		}
	}
}
 
bool cmp(int i,int j){
	return adj[i][0].Y > adj[j][0].Y;
}

int main(){
	ios :: sync_with_stdio(0); cin.tie(0); 
	int n;
	cin >> n;
	rep(i,1,n){
		int u,v,w;
		cin >> u >> v >> w;
		adj[u].pb({v,w});
		adj[v].pb({u,w});
	}
	if (n <= 2){
		cout << 0 << endl;
		return 0;
	}
	dfs(1,0);
	cout << dp[1][0][0] << endl;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...