Submission #366551

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
366551	2021-02-14T13:03:17 Z	YJU	Beads and wires (APIO14_beads)	C++14	100 / 100	466 ms	46940 KB

beads

#include<bits/stdc++.h>
#pragma GCC optimize("unroll-loops,no-stack-protector,Ofast")
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<ll,ll> pll;
const ll MOD=1e9+7;
const ll MOD2=998244353;
const ll N=2e5+5;
const ld pi=acos(-1);
const ll INF=(1LL<<60);
#define SQ(i) ((i)*(i))
#define REP(i,n) for(ll i=0;i<n;i++)
#define REP1(i,n) for(ll i=1;i<=n;i++)
#define pb push_back
#define mp make_pair
#define X first
#define Y second
#define setp setprecision
#define lwb lower_bound
#define SZ(_a) (ll)_a.size()

vector<pll> v[N];
ll n,x,y,c,ans;
multiset<ll,greater<ll> > ms[N];
ll pac[N],dp[N],sum[N];

void DFS(ll nd,ll pa){
	for(auto i:v[nd]){
		if(i.X!=pa){
			pac[i.X]=i.Y;
			DFS(i.X,nd);
			sum[nd]+=dp[i.X];
			ms[nd].insert(sum[i.X]-dp[i.X]+i.Y);
		}
	}
	dp[nd]=max(sum[nd],(SZ(ms[nd])>0?sum[nd]+*ms[nd].begin()+pac[nd]:0LL));
}

void move(ll rt,ll to,ll ed){
	//remove
	sum[rt]-=dp[to];
	ms[rt].erase(ms[rt].find(sum[to]-dp[to]+ed));
	pac[rt]=ed;
	dp[rt]=max(sum[rt],(SZ(ms[rt])>0?sum[rt]+*ms[rt].begin()+pac[rt]:0LL));
	
	sum[to]+=dp[rt];
	pac[to]=0;
	ms[to].insert(sum[rt]-dp[rt]+ed);
	dp[to]=max(sum[to],(SZ(ms[to])>0?sum[to]+*ms[to].begin()+pac[to]:0LL));
}

void reroot(ll nd,ll pa){
	ans=max(ans,sum[nd]);
	for(auto i:v[nd]){
		if(i.X==pa)continue;
		move(nd,i.X,i.Y);
		reroot(i.X,nd);
		move(i.X,nd,i.Y);
	}
}

int main(){
	ios_base::sync_with_stdio(0);cin.tie(0);
	cin>>n;
	REP1(i,n-1){
		cin>>x>>y>>c;
		v[x].pb(mp(y,c));
		v[y].pb(mp(x,c));
	}
	DFS(1,0);
	reroot(1,0);
	cout<<ans<<"\n";
	return 0;
}
/*
10
4 10 2
1 2 21
1 3 13
6 7 1
7 9 5
2 4 3
2 5 8
1 6 55
6 8 34
*/

Subtask #1
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	10 ms	14444 KB	Output is correct
2	Correct	10 ms	14444 KB	Output is correct
3	Correct	10 ms	14444 KB	Output is correct
4	Correct	10 ms	14444 KB	Output is correct
5	Correct	10 ms	14444 KB	Output is correct
6	Correct	10 ms	14444 KB	Output is correct
7	Correct	11 ms	14444 KB	Output is correct
8	Correct	10 ms	14444 KB	Output is correct
9	Correct	10 ms	14444 KB	Output is correct
10	Correct	10 ms	14444 KB	Output is correct
11	Correct	10 ms	14444 KB	Output is correct
12	Correct	10 ms	14444 KB	Output is correct

Subtask #2
15.0 / 15.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	10 ms	14444 KB	Output is correct
2	Correct	10 ms	14444 KB	Output is correct
3	Correct	10 ms	14444 KB	Output is correct
4	Correct	10 ms	14444 KB	Output is correct
5	Correct	10 ms	14444 KB	Output is correct
6	Correct	10 ms	14444 KB	Output is correct
7	Correct	11 ms	14444 KB	Output is correct
8	Correct	10 ms	14444 KB	Output is correct
9	Correct	10 ms	14444 KB	Output is correct
10	Correct	10 ms	14444 KB	Output is correct
11	Correct	10 ms	14444 KB	Output is correct
12	Correct	10 ms	14444 KB	Output is correct
13	Correct	10 ms	14464 KB	Output is correct
14	Correct	10 ms	14444 KB	Output is correct
15	Correct	10 ms	14444 KB	Output is correct
16	Correct	11 ms	14492 KB	Output is correct
17	Correct	10 ms	14444 KB	Output is correct
18	Correct	10 ms	14444 KB	Output is correct
19	Correct	10 ms	14444 KB	Output is correct
20	Correct	10 ms	14444 KB	Output is correct
21	Correct	10 ms	14444 KB	Output is correct
22	Correct	10 ms	14444 KB	Output is correct

Subtask #3
29.0 / 29.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	10 ms	14444 KB	Output is correct
2	Correct	10 ms	14444 KB	Output is correct
3	Correct	10 ms	14444 KB	Output is correct
4	Correct	10 ms	14444 KB	Output is correct
5	Correct	10 ms	14444 KB	Output is correct
6	Correct	10 ms	14444 KB	Output is correct
7	Correct	11 ms	14444 KB	Output is correct
8	Correct	10 ms	14444 KB	Output is correct
9	Correct	10 ms	14444 KB	Output is correct
10	Correct	10 ms	14444 KB	Output is correct
11	Correct	10 ms	14444 KB	Output is correct
12	Correct	10 ms	14444 KB	Output is correct
13	Correct	10 ms	14464 KB	Output is correct
14	Correct	10 ms	14444 KB	Output is correct
15	Correct	10 ms	14444 KB	Output is correct
16	Correct	11 ms	14492 KB	Output is correct
17	Correct	10 ms	14444 KB	Output is correct
18	Correct	10 ms	14444 KB	Output is correct
19	Correct	10 ms	14444 KB	Output is correct
20	Correct	10 ms	14444 KB	Output is correct
21	Correct	10 ms	14444 KB	Output is correct
22	Correct	10 ms	14444 KB	Output is correct
23	Correct	14 ms	15084 KB	Output is correct
24	Correct	14 ms	15084 KB	Output is correct
25	Correct	14 ms	15084 KB	Output is correct
26	Correct	18 ms	15852 KB	Output is correct
27	Correct	19 ms	15828 KB	Output is correct
28	Correct	21 ms	15720 KB	Output is correct
29	Correct	21 ms	15724 KB	Output is correct
30	Correct	21 ms	15724 KB	Output is correct
31	Correct	20 ms	16256 KB	Output is correct

Subtask #4
43.0 / 43.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	10 ms	14444 KB	Output is correct
2	Correct	10 ms	14444 KB	Output is correct
3	Correct	10 ms	14444 KB	Output is correct
4	Correct	10 ms	14444 KB	Output is correct
5	Correct	10 ms	14444 KB	Output is correct
6	Correct	10 ms	14444 KB	Output is correct
7	Correct	11 ms	14444 KB	Output is correct
8	Correct	10 ms	14444 KB	Output is correct
9	Correct	10 ms	14444 KB	Output is correct
10	Correct	10 ms	14444 KB	Output is correct
11	Correct	10 ms	14444 KB	Output is correct
12	Correct	10 ms	14444 KB	Output is correct
13	Correct	10 ms	14464 KB	Output is correct
14	Correct	10 ms	14444 KB	Output is correct
15	Correct	10 ms	14444 KB	Output is correct
16	Correct	11 ms	14492 KB	Output is correct
17	Correct	10 ms	14444 KB	Output is correct
18	Correct	10 ms	14444 KB	Output is correct
19	Correct	10 ms	14444 KB	Output is correct
20	Correct	10 ms	14444 KB	Output is correct
21	Correct	10 ms	14444 KB	Output is correct
22	Correct	10 ms	14444 KB	Output is correct
23	Correct	14 ms	15084 KB	Output is correct
24	Correct	14 ms	15084 KB	Output is correct
25	Correct	14 ms	15084 KB	Output is correct
26	Correct	18 ms	15852 KB	Output is correct
27	Correct	19 ms	15828 KB	Output is correct
28	Correct	21 ms	15720 KB	Output is correct
29	Correct	21 ms	15724 KB	Output is correct
30	Correct	21 ms	15724 KB	Output is correct
31	Correct	20 ms	16256 KB	Output is correct
32	Correct	68 ms	21356 KB	Output is correct
33	Correct	74 ms	21364 KB	Output is correct
34	Correct	70 ms	21356 KB	Output is correct
35	Correct	380 ms	42604 KB	Output is correct
36	Correct	381 ms	42348 KB	Output is correct
37	Correct	378 ms	42348 KB	Output is correct
38	Correct	466 ms	41636 KB	Output is correct
39	Correct	434 ms	41572 KB	Output is correct
40	Correct	435 ms	41556 KB	Output is correct
41	Correct	446 ms	46940 KB	Output is correct

Submission #366551

Compilation message

Subtask #1 13.0 / 13.0

Subtask #2 15.0 / 15.0

Subtask #3 29.0 / 29.0

Subtask #4 43.0 / 43.0

Subtask #1
13.0 / 13.0

Subtask #2
15.0 / 15.0

Subtask #3
29.0 / 29.0

Subtask #4
43.0 / 43.0