【CodeChef】Find a special connected block - CONNECT(斯坦纳树)

时间:2022-11-12 18:21:12

【CodeChef】Find a special connected block - CONNECT(斯坦纳树)

题面

Vjudge

题解

还是一样的套路题,把每个数字映射到\([0,K)\)的整数,然后跑斯坦纳树,重复多次就有很大概率出解。
但是别乱随机,我直接随机\(WA\)成sb了,后来学了别人代码用自己手写的伪随机数就过了。。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
using namespace std;
#define ll long long
#define MAX 16
inline int read()
{
    int x=0;bool t=false;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=true,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return t?-x:x;
}
map<int,int> M;
int n,m,K,ans=2e9;
int a[MAX*MAX],b[MAX*MAX],c[MAX*MAX];
int f[1<<7][MAX*MAX];
queue<int> Q;bool vis[MAX*MAX];
vector<int> E[MAX*MAX];
int ID(int x,int y){return (x-1)*m+y;}
int d[4][2]={-1,0,0,-1,1,0,0,1};
void SPFA(int S)
{
    while(!Q.empty())
    {
        int u=Q.front();Q.pop();
        for(int v:E[u])
            if(f[S][v]>f[S][u]+b[v])
            {
                f[S][v]=f[S][u]+b[v];
                if(!vis[v]&&f[S][v]<=ans)vis[v]=true,Q.push(v);
            }
        vis[u]=false;
    }
}
unsigned int Rand()
{
    static unsigned int x=19260817;
    x^=(x<<5);x^=(x>>17);x^=(x<<13);
    return x;
}
int main()
{
    n=read();m=read();K=read();
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j)a[ID(i,j)]=read();
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j)b[ID(i,j)]=read();
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j)
            if(~a[ID(i,j)])
                for(int k=0;k<4;++k)
                {
                    int x=i+d[k][0],y=j+d[k][1];
                    if(x<1||y<1||x>n||y>m||a[ID(i,j)]==-1)continue;
                    E[ID(i,j)].push_back(ID(x,y));
                }
    for(int Tim=1;Tim<=500;++Tim)
    {
        M.clear();
        for(int i=1;i<=n*m;++i)
            if(~a[i])
            {
                if(!M.count(a[i]))M[a[i]]=Rand()%K;
                c[i]=M[a[i]];
            }
        memset(f,63,sizeof(f));
        for(int i=1;i<=n*m;++i)if(~a[i])f[a[i]?(1<<c[i]):0][i]=b[i];
        for(int S=0;S<(1<<K);++S)
        {
            for(int i=1;i<=n*m;++i)
            {
                if(a[i]==-1)continue;
                for(int T=(S-1)&S;T;T=(T-1)&S)
                    f[S][i]=min(f[S][i],f[T][i]+f[S^T][i]-b[i]);
                if(f[S][i]<=ans)Q.push(i),vis[i]=true;
            }
            SPFA(S);
        }
        for(int i=1;i<=n*m;++i)if(~a[i])ans=min(ans,f[(1<<K)-1][i]);
    }
    printf("%d\n",(ans>1e9)?-1:ans);
    return 0;
}