【BZOJ 4016】[FJOI2014]最短路径树问题

！ 卡时过了 为什么我的这么慢？姿势不对？？？ -->谢 ws_fqk 我的Do（num[i]）应该用 DO(root) 找了半天的root居然没有用....

define的教训永远忘不了了！！！

优化了好几处才过的...者...

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#define MAXN 30010
using namespace std;
int n,m,k;
bool b[MAXN];
int team[10000000],head,tail,d[MAXN];
int tot,g[MAXN],nnext[MAXN*2],num[MAXN*2],cost[MAXN*2];
int tot0,g0[MAXN],nnext0[MAXN*4],num0[MAXN*4],cost0[MAXN*4];
int tot1,g1[MAXN],nnext1[MAXN*4],num1[MAXN*4],cost1[MAXN*4];
struct H{int x,y,z;}G[MAXN*4];
bool cmp(H a,H b){if(a.x==b.x)return a.y>b.y;return a.x<b.x;}
void SPFA()
{
memset(d,61,sizeof(d));
d[1]=0;
team[++tail]=1;
while(head!=tail)
{
int x=team[++head];
b[x]=false;
for(int i=g0[x];i;i=nnext0[i])
{
int tmp=num0[i];
if(d[x]+cost0[i]<d[tmp])
{
d[tmp]=d[x]+cost0[i];
if(!b[tmp])
{
b[tmp]=true;
team[++tail]=tmp;
}
}
}
}
}
void Dfs_Road(int x)
{
b[x]=true;
for(int i=g1[x];i;i=nnext1[i])
{
int tmp=num1[i],z=cost1[i];
if(!b[tmp])
{
tot++,nnext[tot]=g[x],g[x]=tot,num[tot]=tmp,cost[tot]=z;
tot++,nnext[tot]=g[tmp],g[tmp]=tot,num[tot]=x;cost[tot]=z;
Dfs_Road(tmp);
}
}
}
//************点分治***************
int root,sum,max_son[MAXN],size[MAXN],ans0,ans1,N[30000+1][2],T[30000+1][2];

void Get_Root(int x,int fa)
{
size[x]=1;
max_son[x]=0;
for(int i=g[x];i;i=nnext[i])
if(num[i]!=fa&&!b[num[i]])
{
Get_Root(num[i],x);
max_son[x]=max(max_son[x],size[num[i]]);
size[x]+=size[num[i]];
}
max_son[x]=max(max_son[x],sum-size[x]);
if(max_son[x]<max_son[root]) root=x;
}
void Dfs_T(int x,int fa,int depth,int c)
{
if(depth>k) return;
if(c>T[depth][0]) T[depth][0]=c,T[depth][1]=1;
else if(c==T[depth][0]) T[depth][1]++;
for(int i=g[x];i;i=nnext[i])
if(!b[num[i]]&&num[i]!=fa)
Dfs_T(num[i],x,depth+1,c+cost[i]);
}
void Do(int x)
{
for(int i=0;i<=k;i++) N[i][0]=N[i][1]=0;
N[0][1]=1;
b[x]=true;
for(int i=g[x];i;i=nnext[i])
if(!b[num[i]])
{
for(int i=0;i<=k;i++) T[i][0]=T[i][1]=0;
Dfs_T(num[i],-1,1,cost[i]);
for(int i=1;i<=k;i++)
{
if(ans0<N[k-i][0]+T[i][0])
ans0=N[k-i][0]+T[i][0],ans1=N[k-i][1]*T[i][1];
else if(ans0==N[k-i][0]+T[i][0])
ans1+=N[k-i][1]*T[i][1];
}
for(int i=1;i<=k;i++)
{
if(N[i][0]<T[i][0])
N[i][0]=T[i][0],N[i][1]=T[i][1];
else if(N[i][0]==T[i][0])
N[i][1]+=T[i][1];
}
}
for(int i=g[x];i;i=nnext[i])
if(!b[num[i]])
{
root=0;sum=size[num[i]];
Get_Root(num[i],-1);
Do(root);
}
}

int main()
{
cin>>n>>m>>k; k--;
for(int i=1;i<=m;i++)
{
int x,y,z;
scanf("%d %d %d",&x,&y,&z);
G[i]=(H){x,y,z};
G[i+m]=(H){y,x,z};
tot0++;nnext0[tot0]=g0[x],g0[x]=tot0,num0[tot0]=y;cost0[tot0]=z;
tot0++;nnext0[tot0]=g0[y],g0[y]=tot0,num0[tot0]=x,cost0[tot0]=z;
}
SPFA();
sort(G+1,G+m*2+1,cmp);
for(int i=1;i<=m*2;i++)
{
int x=G[i].x,y=G[i].y,z=G[i].z;
if(d[x]+z==d[y])
tot1++,nnext1[tot1]=g1[x],g1[x]=tot1,num1[tot1]=y,cost1[tot1]=z;
}

memset(b,false,sizeof(b));
Dfs_Road(1);

memset(b,false,sizeof(b));
root=0,sum=n,max_son[0]=n+1;
Get_Root(1,-1);
Do(root);

cout<<ans0<<' '<<ans1;
return 0;
}