2019.03.26【CodeChef-CONNECT】Find a special connected block（斯坦纳树）（随机化）

传送门

---

解析：

一杯水，一包烟，一个bug改一天。

我除了一包烟都齐了，TM一个j打成i调了一个晚上

用DZYO的话来说：“这题真TM恶心啊”

显然颜色数过多我们是没有办法做的。

考虑将所有颜色随机映射到$[0,k-1]$中间的整数。

然后对新的矩阵做斯坦纳树。

显然这个做法有一定概率是错的，发现错误的情况只可能是多算了格子，没有区分原来颜色不同的格子。也就是错误情况的答案一定大于正确情况。

发现正确的概率是$\frac{k!}{k^k}$。

那么随机个几百次就差不多了。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define gc get_char
#define cs const

namespace IO{
	inline char get_char(){
		static cs int Rlen=1<<20|1;
		static char buf[Rlen],*p1,*p2;
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;
	}
	
	inline int getint(){
		re char c;
		re bool f=0;
		while(!isdigit(c=gc()))if(c=='-')f=1;re int num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return f?-num:num;
	}
}
using namespace IO;

using std::cout;
using std::cerr;
using std::swap;

cs int N=17,M=1<<7|20,B=1e5+50,INF=1e9;
int T=550;
int n,m,tot,k,bin[N],ans=INF;
#define id(x,y) ((x-1)*m+y)
int f[N*N][M],c[N*N],v[N*N],w[N*N];

int last[N*N],nxt[N*N<<3],to[N*N<<3],ecnt;
inline void addedge(int u,int v){
	nxt[++ecnt]=last[u],last[u]=ecnt,to[ecnt]=v;
}

int q[B],t;
bool in[B];

inline void SPFA(int s){
	t=0;
	for(int re i=1;i<=tot;++i)if(~c[i])
	if(f[i][s]<ans)q[++t]=i,in[i]=true;
	for(int re i=1;i<=t;++i){
		int u=q[i],val=f[u][s];in[u]=false;
		for(int re e=last[u],v=to[e];e;v=to[e=nxt[e]])
		if(~c[v]&&f[v][s]>val+w[v]){
			f[v][s]=val+w[v];
			if(!in[v]&&f[v][s]<ans)in[v]=true,q[++t]=v;
		}
	}
}

inline unsigned int R(){
	static unsigned x=2333;
	x^=x<<5;
	x^=x>>17;
	x^=x<<13;
	return x;
}

inline void solve(){
	for(int re i=1;i<=tot;++i)
	for(int re s=0;s<bin[k];++s)f[i][s]=INF;
	std::map<int,int> id;
	for(int re i=1;i<=tot;++i)if(c[i]>0){
		if(!id.count(c[i]))id[c[i]]=R()%k;
		v[i]=id[c[i]];
	}
	for(int re i=1;i<=tot;++i)if(~c[i])f[i][0]=w[i];
	for(int re s=0;s<bin[k];++s){
		for(int re i=1;i<=tot;++i)if(~c[i]){
			if(c[i]&&(s&bin[v[i]]))f[i][s]=std::min(f[i][s],f[i][s^bin[v[i]]]);
			for(int re j=s;j>s^j;j=s&(j-1))
			f[i][s]=std::min(f[i][s],f[i][j]+f[i][s^j]-w[i]);
		}
		SPFA(s);
	}
	for(int re i=1;i<=tot;++i)ans=std::min(ans,f[i][bin[k]-1]);
}

cs int dx[]={0,0,-1,1};
cs int dy[]={1,-1,0,0};
inline void build(){
	for(int re i=1;i<=n;++i)
	for(int re j=1;j<=m;++j)
	if(~c[id(i,j)])for(int re k=0;k<4;++k){
		int x=i+dx[k],y=j+dy[k];
		if(x<1||x>n||y<1||y>m||!~c[id(x,y)])continue;
		addedge(id(i,j),id(x,y));
	}
}

signed main(){
	n=getint(),m=getint(),k=getint();tot=n*m;
	for(int re i=1;i<=n;++i)
	for(int re j=1;j<=m;++j)
	c[id(i,j)]=getint();
	for(int re i=1;i<=n;++i)
	for(int re j=1;j<=m;++j)
	w[id(i,j)]=getint();
	for(int re i=0;i<=k;++i)bin[i]=1<<i;
	build();
	while(T--)solve();
	cout<<(ans==INF?-1:ans)<<"\n";
	return 0;
}