BZOJ4016: [FJOI2014]最短路径树问题

Description

给一个包含n个点，m条边的无向连通图。从顶点1出发，往其余所有点分别走一次并返回。

往某一个点走时，选择总长度最短的路径走。若有多条长度最短的路径，则选择经过的顶点序列字典序最小的那条路径(如路径A为1,32,11，路径B为1,3,2,11，路径B字典序较小。注意是序列的字典序的最小，而非路径中节点编号相连的字符串字典序最小)。到达该点后按原路返回，然后往其他点走，直到所有点都走过。

可以知道，经过的边会构成一棵最短路径树。请问，在这棵最短路径树上，最长的包含K个点的简单路径长度为多长？长度为该最长长度的不同路径有多少条？

这里的简单路径是指：对于一个点最多只经过一次的路径。不同路径是指路径两端端点至少有一个不同，点A到点B的路径和点B到点A视为同一条路径。

Input

第一行输入三个正整数n,m，K，表示有n个点m条边，要求的路径需要经过K个点。接下来输入m行，每行三个正整数Ai,Bi,Ci(1<=Ai,Bi<=n,1<=Ci<=10000)，表示Ai和Bi间有一条长度为Ci的边。数据保证输入的是连通的无向图。

Output

输出一行两个整数，以一个空格隔开，第一个整数表示包含K个点的路径最长为多长，第二个整数表示这样的不同的最长路径有多少条。

Sample Input

6 6 4
1 2 1
2 3 1
3 4 1
2 5 1
3 6 1
5 6 1

Sample Output

3 4

HINT

对于所有数据n<=30000,m<=60000，2<=K<=n。数据保证最短路径树上至少存在一条长度为K的路径。

Source

点分治

首先我们考虑如何求出最短路径树 我们先找出最短路DAG，然后在最短路DAG上dfs即可

然后点分治统计答案即可

由于这题输出方案 所以就有一些细节

比如更新答案时 要判断方案数*方案数是否为0 否则会统计到错误答案

对于我的代码 即hf[]*hg[]!=0

然后中间的注释就是debug用的了 可以参考

#include<bits/stdc++.h>

using namespace std;

const int maxn=30030;

int n,m,K,cnt,ans,mxdep;

long long tot;

int dist[maxn],head[maxn],size[maxn],dis[maxn],dep[maxn];

int f[maxn],g[maxn],fa[maxn],hf[maxn],hg[maxn];

bool inq[maxn],vis[maxn];

vector <pair<int ,int > > G[maxn];

struct edge
{
	int to,nxt,val;
}e[maxn<<1],E[maxn<<2];

inline void add(int x,int y,int w)
{
	E[++cnt].to=y;
	E[cnt].nxt=head[x];
	head[x]=cnt;
	E[cnt].val=w;
}

inline void Addedge(int x,int y,int w)
{
	e[++cnt].to=y;
	e[cnt].nxt=head[x];
	head[x]=cnt;
	e[cnt].val=w;
}

inline void addedge(int x,int y,int w) { Addedge(x,y,w); Addedge(y,x,w); }

inline void spfa()
{
	memset(dist,0x7f,sizeof(dist));
	queue <int > q;
	q.push(1);
	dist[1]=0;
	while(!q.empty())
	{
		int x=q.front();q.pop();inq[x]=0;
		for(int i=head[x];i;i=E[i].nxt)
		{
			int y=E[i].to;
			if(dist[y]>dist[x]+E[i].val)
			{
				dist[y]=dist[x]+E[i].val;
				if(!inq[y]) inq[y]=1,q.push(y);
			}
		}
	}
	for(int x=1;x<=n;x++)
		for(int i=head[x];i;i=E[i].nxt)
			if(dist[E[i].to]==dist[x]+E[i].val)
				G[x].push_back(make_pair(E[i].to,E[i].val));
	for(int i=1;i<=n;i++) sort(G[i].begin(),G[i].end());
	memset(head,0,sizeof(head));
}

void dfs(int x)
{
	vis[x]=1;
	for(int i=0;i<G[x].size();i++)
		if(!vis[G[x][i].first])
			addedge(x,G[x][i].first,G[x][i].second),dfs(G[x][i].first);
}

void getroot(int x,int Size,int &cg)
{
	bool flag=true;
	size[x]=1;
	for(int i=head[x];i;i=e[i].nxt)
	{
		int y=e[i].to;
		if(!vis[y]&&fa[x]!=y)
		{
			fa[y]=x;
			getroot(y,Size,cg);
			size[x]+=size[y];
			if(size[y]>Size/2) flag=false;
		}
	}
	if(size[x]<Size/2) flag=false;
	if(flag) cg=x;
}

void dfs2(int x)
{
	size[x]=1;
	for(int i=head[x];i;i=e[i].nxt)
	{
		int y=e[i].to;
		if(!vis[y]&&fa[x]!=y)
		{
			fa[y]=x;
			dep[y]=dep[x]+1;
			dis[y]=dis[x]+e[i].val;
			mxdep=max(mxdep,dep[y]);
			if(f[dep[y]]<dis[y])
				f[dep[y]]=dis[y],hf[dep[y]]=1;
			else if(f[dep[y]]==dis[y])
				hf[dep[y]]++;
			dfs2(y);
			size[x]+=size[y];
		}
	}
}

void solve(int x,int Size)
{
//	printf("-------------------------------------------\n");
	int cg,mx=0;
	f[x]=0;
	getroot(x,Size,cg);
	vis[cg]=1;size[fa[cg]]=Size-size[cg];fa[cg]=0;
	hg[0]=1;
//	printf("[%d]\n",cg);
	for(int i=head[cg];i;i=e[i].nxt)
		if(!vis[e[i].to])
		{
			dis[e[i].to]=e[i].val;
			dep[e[i].to]=1;
			fa[e[i].to]=cg;
			mxdep=1;
			f[1]=e[i].val,hf[1]++;
			dfs2(e[i].to);
			mx=max(mx,mxdep);
			for(int j=0;j<=mxdep;j++)
			{
				if(hf[j]&&hg[K-1-j])
				{
					if(ans<f[j]+g[K-1-j]) 
						ans=f[j]+g[K-1-j],tot=1ll*hf[j]*hg[K-1-j];
					else if(ans==f[j]+g[K-1-j])
						tot+=1ll*hf[j]*hg[K-1-j];
//					cout<<ans<<" "<<tot<<" "<<j<<" "<<f[j]<<" "<<g[K-1-j]<<" "<<hf[j]<<" "<<hg[K-1-j]<<endl;
				}
			}
//			putchar(10);
			for(int j=0;j<=mxdep;j++)
			{
				if(f[j]>g[j])
					g[j]=f[j],hg[j]=hf[j];
				else if(f[j]==g[j])
					hg[j]+=hf[j];
				f[j]=0,hf[j]=0;
			}
		}
	for(int i=0;i<=mx;i++) g[i]=0,hg[i]=0;
	for(int i=head[cg];i;i=e[i].nxt) if(!vis[e[i].to]) solve(e[i].to,size[e[i].to]);
}

int main()
{
	scanf("%d%d%d",&n,&m,&K);
	for(int i=1;i<=m;i++)
	{
		int x,y,w;
		scanf("%d%d%d",&x,&y,&w);
		add(x,y,w);add(y,x,w);
	}
	spfa();
	dfs(1);
	memset(vis,0,sizeof(vis));
	solve(1,n);
	return printf("%d %lld\n",ans,tot),0;
}