JZOJ3400-旅行

Description&Data Constraint

从前有一位旅者，他想要游遍天下所有的景点。这一天他来到了一个神奇的王国：在这片土地上，有n个城市，从1到n进行编号。王国中有m条道路，第i条道路连接着两个城市ai,bi，由于年代久远，所有的道路都已经不能使用。如果要修复第i条道路，需要wi的时间。为了更好的旅行，旅者想要将某些道路修复，使得1号城市能够到达n号城市，2号城市能够到达n-1号城市…k号城市能够到达n-k+1号城市。为了满足他的要求，请问最少需要多少时间去修复道路。无解请输出-1。

$k\le4, 2k\le n\le10^4, m\le10^4, wi\le10^3, 1 \le a_i, b_i \le n $

Solution

正解是不完全的斯坦纳树板子题

但是我不会！准确来说是能理解但不会讲

所以我来讲水法（JZOJ可以过）

由于$k$比较小，可以全排列

然后每个走最短路，走过的$val$改为0，可以通过$pre$来实现

最后在全排列中花费最少的就是答案

当然是很容易被hack的

所以还是去学斯坦纳树吧！

Code

1. 水法

#include<cstdio>
#include<cstring>
#define N 10005
#define inf 1234567890
using namespace std;
struct node
{
	int head,next,to,val,flag;
}a[N*2],save[N*2];
int n,m,k,x,y,z,tot,ans,res,num,c[5],dis[N],d[N<<2],g[N];
bool B[5],b[N],bj[N],bz[N];
void add(int x,int y,int z,int Flag)
{
	a[++tot].to=y;
	a[tot].val=z;
	a[tot].flag=Flag;
	a[tot].next=a[x].head;
	a[x].head=tot;
}
int spfa(int x,int y)
{
	int h=0,t=1;
	d[1]=x;g[x]=0;bj[x]=true;dis[x]=0;
	while (h<t)
	{
		int u=d[++h];
		for (int i=a[u].head;i;i=a[i].next)
		{
			int v=a[i].to;
			if (dis[u]+a[i].val<dis[v])
			{
				dis[v]=dis[u]+a[i].val;
				if (a[i].flag==1) g[v]=i+1;
				else g[v]=i-1;
				if (!bj[v]) d[++t]=v,bj[v]=true;
			}
		}
		bj[u]=false;
	}
	return dis[y];
}
void dg(int x)
{
	if (x>k)
	{
		res=0;
		for (int i=1;i<=m*2;++i)
		{
			save[i].flag=a[i].flag;
			save[i].head=a[i].head;
			save[i].next=a[i].next;
			save[i].to=a[i].to;
			save[i].val=a[i].val;
		}
		for (int i=1;i<=k;++i)
		{
			memset(bj,false,sizeof(bj));
			memset(d,0,sizeof(d));
			memset(g,0,sizeof(g));
			memset(dis,0X3f3f3f3f,sizeof(dis));
			res+=spfa(c[i],n-c[i]+1);
			int l=g[n-c[i]+1];
			while (l)
			{
				a[l].val=a[l+a[l].flag].val=0;
				l=g[a[l].to];
			}
		}
		if (res<ans) ans=res;
		for (int i=1;i<=m*2;++i)
		{
			a[i].flag=save[i].flag;
			a[i].head=save[i].head;
			a[i].next=save[i].next;
			a[i].to=save[i].to;
			a[i].val=save[i].val;
		}
		return;
	}
	for (int i=1;i<=k;++i)
	{
		if (!B[i])
		{
			c[x]=i;
			B[i]=true;
			dg(x+1);
			B[i]=false;
			c[x]=0;	
		}
	}
}
int main()
{
	scanf("%d%d%d",&n,&m,&k);
	for (int i=1;i<=m;++i)
	{
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z,1);add(y,x,z,-1);
	}
	ans=inf;
	dg(1);
	if (ans==inf) printf("-1\n");
	else printf("%d\n",ans);
	return 0;
}

2. 正解

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm> 
#define inf 0X3f3f3f3f
#define N 10005
using namespace std;
struct node
{
	int to,next,head,val;
}a[N*2];
queue <int> q;
int n,m,k,x,y,z,tot,f[N][260],ans[260],res[260];
bool b[N];
void add(int x,int y,int z)
{
	a[++tot].to=y;
	a[tot].val=z;
	a[tot].next=a[x].head;
	a[x].head=tot;
}
void spfa(int s)
{
	memset(b,true,sizeof(b));
	while (!q.empty())
	{
		int u=q.front();
		q.pop();
		for (int i=a[u].head;i;i=a[i].next)
		{
			int v=a[i].to;
			if (f[v][s]>f[u][s]+a[i].val)
			{
				f[v][s]=f[u][s]+a[i].val;
				if (!b[v])
				{
					q.push(v);
					b[v]=true;
				}
			}
		}
		b[u]=false;
	}
}
bool pd(int x)
{
	for (int i=1;i<=k;++i)
	{
		int b1=x&(1<<(i-1)),b2=x&(1<<(k+i-1));
		if ((!b1&&b2)||(b1&&!b2)) return false;
	}
	return true;
}
int main()
{
	memset(f,inf,sizeof(f));
	memset(res,inf,sizeof(res));
	memset(ans,inf,sizeof(ans));
	scanf("%d%d%d",&n,&m,&k);
	for (int i=1;i<=m;++i)
	{
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);add(y,x,z);
	}
	for (int i=1;i<=k;++i)
		f[i][1<<(i-1)]=f[n-i+1][1<<(k+i-1)]=0;
	for (int s=1;s<(1<<(2*k));++s)
	{
		for (int i=1;i<=n;++i)
		{
			for (int ss=s&(s-1);ss;ss=s&(ss-1)) 
					f[i][s]=min(f[i][s],f[i][ss]+f[i][s^ss]);
			q.push(i);
		}
		spfa(s);
		for (int i=1;i<=n;++i)
			res[s]=min(res[s],f[i][s]);
	}
	ans[0]=0;
	for (int s=1;s<(1<<(2*k));++s)
	{
		if (!pd(s)) continue;
		for (int ss=0;ss<s;++ss)
			if ((ss|s)==s) ans[s]=min(ans[s],ans[ss]+res[s^ss]);
	}
	if (ans[(1<<(2*k))-1]>=inf) printf("-1\n");
	else printf("%d\n",ans[(1<<(2*k))-1]);
	return 0;
}