【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
#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);
}