斯坦纳树题集

[WC2008]游览计划

网格图，点权，连通一些指定点的最小代价【输出方案】。

Code：

#include<bits/stdc++.h>
#define maxn 12
#define maxs (1<<10)
using namespace std;
const int inf = 0x3f3f3f3f;
int T,n,m,k,a[maxn][maxn],f[maxn][maxn][maxs];
struct node{int x,y,s;}pre[maxn][maxn][maxs];
struct Point{int x,y;};
queue<Point>q; bool inq[maxn][maxn];
int dx[4]={1,-1,0,0},dy[4]={0,0,1,-1};
bool vis[maxn][maxn];
void dfs(int i,int j,int s){
	vis[i][j]=1; node tmp=pre[i][j][s];
	if(!tmp.s) return;
	dfs(tmp.x,tmp.y,tmp.s);
	if(s!=tmp.s) dfs(tmp.x,tmp.y,s^tmp.s);
}
int main()
{
	int x,y;
	scanf("%d%d",&n,&m);
	memset(f,0x3f,sizeof f);
	for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) scanf("%d",&a[i][j]),!a[i][j]&&(f[i][j][1<<(k++)]=0);
	for(int s=1;s<1<<k;s++){
		for(int i=1;i<=n;i++)
			for(int j=1;j<=m;j++){
				for(int t=(s-1)&s;t;t=(t-1)&s)
					if(f[i][j][t]+f[i][j][s^t]-a[i][j]<f[i][j][s])
						f[i][j][s]=f[i][j][t]+f[i][j][s^t]-a[i][j],pre[i][j][s]=(node){i,j,t};
				if(f[i][j][s]<inf) inq[i][j]=1,q.push((Point){i,j});
			}
		while(!q.empty()){
			int u=q.front().x,v=q.front().y; q.pop(),inq[u][v]=0;
			for(int i=0;i<4;i++)
				if((x=u+dx[i])>=1&&x<=n&&(y=v+dy[i])>=1&&y<=m&&f[x][y][s]>f[u][v][s]+a[x][y]){
					f[x][y][s]=f[u][v][s]+a[x][y],pre[x][y][s]=(node){u,v,s};
					if(!inq[x][y]) inq[x][y]=1,q.push((Point){x,y});
				}
		}
	}
	for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) if(!a[i][j]) {x=i,y=j;goto end;}
end:
	printf("%d\n",f[x][y][(1<<k)-1]);
	dfs(x,y,(1<<k)-1);
	for(int i=1;i<=n;i++,puts("")) for(int j=1;j<=m;j++) putchar(vis[i][j]?a[i][j]?'o':'x':'_');
}

HDU4085 Peach Blossom Spring

无向带边权图，将$[1,k]$与$[n-k+1,n]$配对连通的最小总代价。【斯坦纳森林，枚举子集合并答案，注意合并的状态必须是合法状态】。

Code：

#include<bits/stdc++.h>
#define maxn 52
#define maxm 2005
using namespace std;
const int inf = 0x3f3f3f3f;
int T,n,m,k,f[maxn][1<<10],g[1<<10],bin[1<<10];
int fir[maxn],nxt[maxm],to[maxm],w[maxm],tot;
inline void line(int x,int y,int z){nxt[++tot]=fir[x],fir[x]=tot,to[tot]=y,w[tot]=z;}
queue<int>q; bool inq[maxn];
int main()
{
	for(int i=1;i<1<<10;i++) bin[i]=bin[i>>1]+(i&1);
	scanf("%d",&T);
	while(T--){
		scanf("%d%d%d",&n,&m,&k); int x,y,z;
		memset(fir,0,(n+1)<<2),tot=0;
		for(int i=1;i<=m;i++) scanf("%d%d%d",&x,&y,&z),line(x,y,z),line(y,x,z);
		for(int i=1;i<=n;i++) for(int s=0;s<1<<2*k;s++) f[i][s]=inf;
		for(int i=1;i<=k;i++) f[i][1<<(i-1)]=0,f[n-i+1][1<<(2*k-i)]=0;
		for(int s=1;s<1<<2*k;s++){
			for(int i=1;i<=n;i++){
				for(int t=(s-1)&s;t;t=(t-1)&s)
					f[i][s]=min(f[i][s],f[i][t]+f[i][s^t]);
				if(f[i][s]<inf) inq[i]=1,q.push(i);
			}
			while(!q.empty()){
				int u=q.front(); q.pop(),inq[u]=0;
				for(int i=fir[u],v;i;i=nxt[i]) if(f[v=to[i]][s]>f[u][s]+w[i]){
					f[v][s]=f[u][s]+w[i]; if(!inq[v]) inq[v]=1,q.push(v);
				}
			}
		}
		memset(g,0x3f,(1<<2*k)<<2);
		for(int s=1;s<1<<2*k;s++) if(bin[s&((1<<k)-1)]==bin[s>>k]){
			for(int i=1;i<=n;i++) g[s]=min(g[s],f[i][s]);
			for(int t=(s-1)&s;t;t=(t-1)&s) g[s]=min(g[s],g[t]+g[s^t]);
		}
		if(g[(1<<2*k)-1]<inf) printf("%d\n",g[(1<<2*k)-1]);
		else puts("No solution");
	}
}

ZOJ3613 Wormhole Transport

无向带边权图，点上可能有一个资源点/多个工厂，将资源点与工厂配对，求最大配对数及最小代价。【同一个点上的工厂可以合在一个连通状态上，合并答案时的状态要求工厂数大于等于资源点数】。

Code：

#include<bits/stdc++.h>
#define maxn 205
#define maxm 10005
using namespace std;
const int inf = 0x3f3f3f3f;
int T,n,m,k,S[maxn],X[maxn],cx,cy,f[maxn][1<<8],g[1<<8],bin[1<<8];
int fir[maxn],nxt[maxm],to[maxm],w[maxm],tot;
inline void line(int x,int y,int z){nxt[++tot]=fir[x],fir[x]=tot,to[tot]=y,w[tot]=z;}
queue<int>q; bool inq[maxn];
inline bool check(int s){
	int ret=0;
	for(int i=0;i<cx;i++) if(s>>i&1) ret+=X[i];
	for(int i=cx;i<cx+cy;i++) if(s>>i&1) ret--;
	return ret>=0;
}
int main()
{
	for(int i=1;i<1<<8;i++) bin[i]=bin[i>>1]+(i&1);
	int x,y,z,sum;
	while(~scanf("%d",&n)){
		memset(fir,0,(n+1)<<2),tot=0,cx=cy=sum=0;
		memset(f,0x3f,sizeof f);
		for(int i=1;i<=n;i++){
			scanf("%d%d",&x,&S[i]);
			if(x&&S[i]) sum++,x--,S[i]=0;
			if(x) f[i][1<<cx]=0,X[cx++]=x; 
		}
		for(int i=1;i<=n;i++) if(S[i]) f[i][1<<(cx+cy)]=0,cy++;
		scanf("%d",&m);
		for(int i=1;i<=m;i++) scanf("%d%d%d",&x,&y,&z),line(x,y,z),line(y,x,z);
		for(int s=1;s<1<<(cx+cy);s++){
			for(int i=1;i<=n;i++){
				for(int t=(s-1)&s;t;t=(t-1)&s)
					f[i][s]=min(f[i][s],f[i][t]+f[i][s^t]);
				if(f[i][s]<inf) inq[i]=1,q.push(i);
			}
			while(!q.empty()){
				int u=q.front(); q.pop(),inq[u]=0;
				for(int i=fir[u],v;i;i=nxt[i]) if(f[v=to[i]][s]>f[u][s]+w[i]){
					f[v][s]=f[u][s]+w[i]; if(!inq[v]) inq[v]=1,q.push(v);
				}
			}
		}
		memset(g,0x3f,sizeof g);
		int ans=0,cnt=0;
		for(int s=1;s<1<<(cx+cy);s++) if(check(s)){
			for(int i=1;i<=n;i++) g[s]=min(g[s],f[i][s]);
			for(int t=(s-1)&s;t;t=(t-1)&s) g[s]=min(g[s],g[t]+g[s^t]);
			int num=bin[s>>cx];
			if(num>cnt) ans=g[s],cnt=num;
			else if(num==cnt) ans=min(ans,g[s]);
		}
		printf("%d %d\n",cnt+sum,ans);
	}
}

模拟赛20200302 LYK loves graph

至少连通k种关键点。【随机映射】https://blog.csdn.net/C20181220_xiang_m_y/article/details/104615822

HDU3311 Dig The Wells

【打井问题】n+m个点每个点打井有个费用，连边有费用，问让前n个点都与井相连的最小费用。

新建一个虚点0，每个点向它连边，费用为打井的费用，问题转化为求让0和前n个点连通的最小费用。

Code：

#include<bits/stdc++.h>
#define maxn 1015
#define maxm 11015
using namespace std;
const int inf = 0x3f3f3f3f;
int n,m,E,f[maxn][1<<5];
int fir[maxn],nxt[maxm],to[maxm],w[maxm],tot;
void line(int x,int y,int z){nxt[++tot]=fir[x],fir[x]=tot,to[tot]=y,w[tot]=z;}
queue<int>q; bool inq[maxn];
int main()
{
    for(int x,y,z;~scanf("%d%d%d",&n,&m,&E);){
        memset(fir,0,(n+m+1)<<2),tot=0;
        for(int i=1;i<=n;i++) scanf("%d",&x),line(i,0,x);
        for(int i=1;i<=m;i++) scanf("%d",&x),line(n+i,0,x);
        for(int i=1;i<=E;i++) scanf("%d%d%d",&x,&y,&z),line(x,y,z),line(y,x,z);
        memset(f,0x3f,(n+m+1)*(1<<5)<<2);
        for(int i=1;i<=n;i++) f[i][1<<(i-1)]=0;
        for(int s=1;s<1<<n;s++){
            for(int i=0;i<=n+m;i++){
                for(int t=s-1;t;t=(t-1)&s) f[i][s]=min(f[i][s],f[i][t]+f[i][s^t]);
                if(f[i][s]<inf) q.push(i),inq[i]=1;
            }
            while(!q.empty()){
                int u=q.front(); q.pop(),inq[u]=0;
                for(int i=fir[u],v;i;i=nxt[i]) if(f[v=to[i]][s]>f[u][s]+w[i]){
                    f[v][s]=f[u][s]+w[i]; if(!inq[v]) inq[v]=1,q.push(v);
                }
            }
        }
        printf("%d\n",f[0][(1<<n)-1]);
    }
}

LOJ#2977. 「THUSCH 2017」巧克力

网格图，每个点有颜色和权值，求含有至少$k\le 5$种颜色的连通块的最少点数，在此前提下求连通块权值中位数的最小值。【随机映射】【二分求最小中位数】

随机映射到$k$种颜色，某种最优解恰好被映射到不同的$k$种颜色的概率是$\frac{k!}{k^k}=0.0384$，随机两百次还不成功的概率是$3.970e-4$

先求出连通块的最少点数，记为$cnt$
然后二分一个数$mid$，检验是否可以在最少点数的条件下尽可能多的选择小于等于$mid$的数，使其达到$\frac{cnt+1}{2}$。
令每个数的权值为$a[i]=B+(a[i]\le mid?-1:1)$。（$B$是一个大于$cnt$的数，可以自己设定，只要后面的值的和达不到$B$）
如果这样算出连通$k$种颜色的最小权值小于等于$cnt\ast B$，说明当前的中位数还可以更小，令$r=mid$，否则令$l=mid+1$。
原理就是给第一条件设一个大权值，第二条件设一个小权值，求最小就会在优先第一条件的情况下尽可能使第二条件最小。

Code：

#include<bits/stdc++.h>
#define maxn 235
#define maxk 5
using namespace std;
const int inf = 0x3f3f3f3f;
int n,m,k,N,f[maxn][1<<maxk],a[maxn],A[maxn],b[maxn],c[maxn],id[maxn],len,ansP,ansV;
int dx[4]={1,-1,0,0},dy[4]={0,0,1,-1};
typedef pair<int,int> pii;
priority_queue<pii,vector<pii>,greater<pii> >q;
const int B = 1000;
int Steiner(int mid){
	memset(f,0x3f,N*(1<<maxk)<<2);
	for(int i=0;i<N;i++) if(~c[i]) f[i][1<<id[c[i]]]=A[i]=B+(a[i]<=mid?-1:1);
	for(int s=1;s<1<<k;s++){
		for(int i=0;i<N;i++){
			for(int t=s-1;t;t=(t-1)&s) f[i][s]=min(f[i][s],f[i][t]+f[i][s^t]-A[i]);
			if(f[i][s]<inf) q.push(pii(f[i][s],i));
		}
		while(!q.empty()){
			int u=q.top().second,d=q.top().first,i=u/m,j=u%m; q.pop();
			if(f[u][s]!=d) continue;
			for(int k=0,x,y,v;k<4;k++) 
				if((x=i+dx[k])>=0&&x<n&&(y=j+dy[k])>=0&&y<m&&~c[v=x*m+y]&&f[v][s]>f[u][s]+A[v])
					q.push(pii(f[v][s]=f[u][s]+A[v],v));
		}
	}
	int ret=inf;
	for(int i=0;i<N;i++) ret=min(ret,f[i][(1<<k)-1]);
	return ret;
}
void solve(){
	for(int i=1;i<=N;i++) id[i]=rand()%k;
	int cnt=Steiner(-1);
	if(cnt>=inf||ansP<(cnt/=B)) return;
	if(ansP>cnt) ansP=cnt,ansV=inf;
	int l=0,r=min(len-1,ansV-1),mid;
	if(Steiner(b[r])>cnt*B) return;
	while(l<r) mid=(l+r)>>1,Steiner(b[mid])<=cnt*B?(r=mid):(l=mid+1);
	ansV=l;
}
int main()
{
	srand(time(0));
	int T; scanf("%d",&T);
	while(T--){
		scanf("%d%d%d",&n,&m,&k),N=n*m,len=0;
		for(int i=0;i<N;i++) scanf("%d",&c[i]);
		for(int i=0;i<N;i++) scanf("%d",&a[i]),~c[i]&&(b[len++]=a[i]);
		sort(b,b+len),len=unique(b,b+len)-b-1;
		int S = 150; ansP=inf,ansV=0;
		while(S--) solve();
		if(ansP<inf) printf("%d %d\n",ansP,b[ansV]);
		else puts("-1 -1");
	}
}