【JZOJ B組】【GDSOI2017第二輪模擬】奶酪

題目

Description
CJY很喜歡吃奶酪，因而YJC弄到了一些奶酪，如今YJC決定和CJY分享奶酪。
YJC弄到了n-1塊奶酪，因而他把奶酪掛在了一棵n個結點的樹上，每根樹枝上掛一塊奶酪，每塊奶酪都有重量。
YJC和CJY決定這樣分奶酪：首先砍掉一根樹枝，把樹分紅兩部分，每人取一部分，而後各自在本身取的那部分樹上選擇一條路徑並取走路徑上的奶酪，而後把剩下的奶酪拿去喂老鼠。
兩人都想讓本身取走總重量儘可能大的奶酪，但他們不知道砍掉哪一根樹枝最好。因此他們想讓你計算，對於每一根樹枝，砍掉以後每一個人取走的奶酪的總重量的最大值。

Input
第一行一個正整數n，表示樹的點數。
接下來的n-1行每行有三個數u,v,w，表示有一根樹枝鏈接u和v，掛在上面的奶酪的重量爲w。

Output
對於第i根樹枝(從1開始標號)，你將會獲得2個答案Pi和Qi，咱們令ansi=max(Pi,Qi)23333+min(Pi,Qi)2333+233ii+23*i+2，爲了縮短輸出答案時間，請只輸出一個數S表示全部ansi的和對2333333333333333取模的值。

Sample Input
10
1 2 234
2 9 936
9 5 784
5 3 105
2 8 775
8 10 368
10 6 1003
9 4 670
4 7 417

Sample Output
735923484

Data Constraint
對於10%的數據，保證n<=310^2。
對於30%的數據，保證n<=510^3
對於50%的數據，保證n<=10^5
對於60%的數據，保證n<=410^5
對於70%的數據，保證n<=10^6
對於100%的數據，保證n<=410^6,w<=106

思路

一個比較經典的樹形DP，主要處理好子樹的幾個最大值
可是我在考場上只拿了10分，一開始找不到緣由，結果是i沒開longlong見了祖宗……

代碼

#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll inf=0x7fffffff,N=4e6+77,mod=2333333333333333;
ll n,bz[N],id,ls[N],to[N*2],next[N*2],val[N*2],cnt=1,tmb,ban;
ll f[N],g[N],h[N],yjy[N],yjyx[N],yjyxx[N],faii[N],F[N],aii[N];
ll ans,ans1,ans2;
struct E
{
	int to,next,v;
}e[N*2];
void add(ll u,ll v,ll val)
{
	e[++cnt].to=v; e[cnt].next=ls[u]; ls[u]=cnt; e[cnt].v=val;
}
void dfs(ll u,ll fa)
{
	for(int i=ls[u]; i; i=e[i].next)
		if(e[i].to!=fa)
		{
			faii[e[i].to]=u;
			dfs(e[i].to,u);
			yjy[u]=max(yjy[e[i].to],yjy[u]);
			ll tmp=f[e[i].to]+e[i].v;
			if(tmp>=f[u])
			{
				h[u]=g[u];
				g[u]=f[u];
				f[u]=tmp;
			}
			else if(tmp>=g[u])
			{
				h[u]=g[u];
				g[u]=tmp;
			}
			else if(tmp>=h[u]) h[u]=tmp;
			if(yjy[e[i].to]>=yjyx[u])
			{
				yjyxx[u]=yjyx[u];
				yjyx[u]=yjy[e[i].to];
			}
			else if(yjy[e[i].to]>=yjyxx[u]) yjyxx[u]=yjy[e[i].to];
		}
	yjy[u]=max(yjy[u],f[u]+g[u]);
}
void dfs2(ll u,ll fa)
{
	for(int i=ls[u]; i; i=e[i].next)
		if(e[i].to!=fa)
		{
			aii[e[i].to]=aii[u];
			if(yjy[e[i].to]==yjyx[u]) aii[e[i].to]=max(aii[e[i].to],yjyxx[u]);
			else aii[e[i].to]=max(aii[e[i].to],yjyx[u]);
			if(f[u]==f[e[i].to]+e[i].v)
			{
				F[e[i].to]=max(F[u]+e[i].v,g[u]+e[i].v);
				aii[e[i].to]=max(aii[e[i].to],g[u]+max(h[u],F[u]));
			}
			else
			{
				F[e[i].to]=max(F[u]+e[i].v,f[u]+e[i].v);
				if(g[u]==f[e[i].to]+e[i].v) aii[e[i].to]=max(aii[e[i].to],f[u]+max(h[u],F[u]));
				else aii[e[i].to]=max(aii[e[i].to],f[u]+max(g[u],F[u]));
			}
			dfs2(e[i].to,u);
		}
}

int main()
{
	freopen("cheese.in","r",stdin); freopen("cheese.out","w",stdout);
	scanf("%d",&n);
	for(int i=1,x,y,z; i<n; i++) scanf("%d%d%d",&x,&y,&z),add(x,y,z),add(y,x,z);
	dfs(1,0);
	dfs2(1,0);
	for(ll i=1; i<n; i++)
	{
		if(i==3036)
		{
// printf("***\n");
		}
		ll u=e[i*2+1].to,v=e[i*2].to;
		if(faii[v]==u) swap(u,v);
		ans1=yjy[u];
		ans2=aii[u];
		ans=(ans+max(ans1,ans2)*23333+min(ans2,ans1)*2333+233*i*i+23*i+2)%mod;
// printf("%lld %lld %lld\n",ans1,ans2,ans);
	}
	printf("%lld",ans);
}