斯坦纳树入门

斯坦纳树可以解决下面的问题
给定一个带权无向图和某些关键点,求一棵最小生成树(必须满足所有关键节点都在此树上)
考虑使用状压DP求解
f[i][s]表示以i为根节点的树中选择关键点的状态为s(状压)
枚举一个t(t|s==s),f[i][s]=min(f[i][s],f[i][t]+f[i][s-t])
val是i与son连边的代价
看起来很好理解,就是将两棵树拼成一棵新树。
还有第二种转移:f[i][s]=min(f[i][s],f[son][s]+val),val是连边代价
这也没什么好说的,发现这种形式有后效性,可以用最短路去实现。
然后问题就解决了。
例题1:JZOJ3400 旅行
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int i,j,k,m,n,o,p,l,s,t,x,y,z,r,ll;
int dp[10001][501],ans[501],good[501];
int f[100001][4],q[100001],h[100001],bz[10001];
void insert(int x,int y,int z)
{ f[++t][1]=y,f[t][2]=q[x],f[t][3]=z,q[x]=t; } 
int pd(int s)
{
	for (int i=1;i<=ll;i++)
	{
		int gg=s&(1<<(i-1)),hh=s&(1<<(ll+i-1));
		if ((!gg&&hh)||(!hh&&gg)) return 0;
	}
	return 1;
}
int main()
{
	freopen("travel.in","r",stdin);
	freopen("travel.out","w",stdout);
	scanf("%d%d%d",&n,&m,&ll);
	for (i=1;i<=m;i++) scanf("%d%d%d",&x,&y,&z),insert(x,y,z),insert(y,x,z);
	int N=(1<<(2*ll))-1;
	memset(dp,60,sizeof(dp));memset(good,60,sizeof(good));memset(ans,60,sizeof(ans));
	for (i=1;i<=ll;i++)
		dp[i][1<<(i-1)]=dp[n-i+1][1<<(ll+i-1)]=0;
	for (s=1;s<=N;s++)
	{
		for (i=1;i<=n;i++)
		{
			for (t=1;t<=s;t++)
				if ((t|s)==s)
				{
					for (k=q[i];k;k=f[k][2])
						dp[i][s]=min(dp[i][s],dp[i][t]+dp[i][s-t]);
				}
		}
		l=0,r=n;memset(bz,0,sizeof(bz));
		for (j=1;j<=n;j++) h[j]=j,bz[j]=1;
		while (l<r)
		{
			l++;
			for (int k=q[h[l]];k;k=f[k][2])
			{
				if (dp[h[l]][s]+f[k][3]<dp[f[k][1]][s])
				{
					dp[f[k][1]][s]=dp[h[l]][s]+f[k][3];
					if (!bz[f[k][1]]) bz[f[k][1]]=1,h[++r]=f[k][1];	
				}	
			}	
			bz[h[l]]=0;
		} 
		for (i=1;i<=n;i++) good[s]=min(good[s],dp[i][s]);
	}
	ans[0]=0;
	for (s=1;s<=N;s++)
	{
		if (s==N)
		{
			int gds=0;
		 } 
		for (t=0;t<s;t++)
			if ((t|s)==s&&pd(s)) ans[s]=min(ans[s],ans[t]+good[s-t]);
	}
	if (ans[N]>1e9) ans[N]=-1;
	printf("%d\n",ans[N]);
}
例题2:JZOJ3737. 【NOI2014模拟7.11】挖宝藏 (treasure)
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int f[11][11][11][2049],a[11][11][11],b[11][11][11];
int h[100001][2],c[11],spfa[11][11],tr[11][11][11];
int way[5][3];
int i,j,k,m,n,o,p,l,s,t,H,r,ans;
void Shortest_Path_Faster_Algorithm()//SPFA
{
	while (l<r)
	{
		l++;
		int x=h[l][0],y=h[l][1];
		for (int k=1;k<=4;k++)
		{
			int xx=x+way[k][1],yy=y+way[k][2];
			if (xx<=0||yy<=0||xx>n||yy>m) continue;
			if (f[i][x][y][s]+a[i][xx][yy]<f[i][xx][yy][s])
			{
				f[i][xx][yy][s]=f[i][x][y][s]+a[i][xx][yy];
				if (!spfa[xx][yy]) spfa[xx][yy]=true,h[++r][0]=xx,h[r][1]=yy;
			}
		}
		spfa[x][y]=false;
	}
}
int main()
{
	freopen("treasure.in","r",stdin);
	freopen("treasure.out","w",stdout);
	scanf("%d%d%d",&H,&n,&m);
	for (i=1;i<=H;i++)
		for (j=1;j<=n;j++)
			for (k=1;k<=m;k++)
				scanf("%d",&a[i][j][k]);
	for (i=1;i<=H;i++)
	{
		scanf("%d",&c[i]);
		for (j=1;j<=c[i];j++) scanf("%d%d",&b[i][j][0],&b[i][j][1]),tr[i][b[i][j][0]][b[i][j][1]]=j;
		c[i]+=(i>1);//对于第i层(i!=1),多开一个宝藏位
	}
	memset(f,60,sizeof(f));way[1][1]=1,way[1][2]=0,way[2][1]=0,way[2][2]=1,way[3][1]=-1,way[3][2]=0,way[4][1]=0,way[4][2]=-1;
	for (i=1;i<=H;i++)
	{
		if (i==2)
		{
			int gawra=0;
		}
		for (j=1;j<=n;j++)
			for (k=1;k<=m;k++)
				if (tr[i][j][k]) f[i][j][k][1<<(tr[i][j][k]-1)]=a[i][j][k];
				else f[i][j][k][0]=a[i][j][k];
		for (s=0;s<=(1<<c[i])-1;s++)
		{
			l=0,r=0;
			for (j=1;j<=n;j++)
				for (k=1;k<=m;k++)
				{
					for (t=s&(s-1);t;t=s&(t-1))
							f[i][j][k][s]=min(f[i][j][k][s],f[i][j][k][t]+f[i][j][k][s-t]-a[i][j][k]);	
					if (f[i][j][k][s]<1e9) h[++r][0]=j,h[r][1]=k,spfa[j][k]=true;
				}
			Shortest_Path_Faster_Algorithm();
			if (s==(1<<c[i])-1)
			{
				for (j=1;j<=n;j++)
					for (k=1;k<=m;k++)
					{
						if (f[i][j][k][s]>1e9||(i==H)) continue;
						if (tr[i+1][j][k])
							f[i+1][j][k][(1<<(c[i+1]-1))|(1<<(tr[i+1][j][k]-1))]=f[i][j][k][s]+a[i+1][j][k];//注意因为下一层对应位可能有宝藏,要与“超级宝藏”异或一下
						else 
							f[i+1][j][k][(1<<(c[i+1]-1))]=f[i][j][k][s]+a[i+1][j][k];//将此层答案转移到下一层的“超级宝藏”
					}
			}
		}	
	}
	ans=1e9,s=(1<<c[H])-1;
	for (i=1;i<=n;i++)
		for (j=1;j<=m;j++)
			ans=min(ans,f[H][i][j][s]);
	printf("%d\n",ans);
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值