bzoj 1556: 墓地秘密状压dp+spfa

Posted 2021-01-04 lokiii

显然是状压，显然不可能把所有格子压起来
仔细观察发现只有机关周围的四个格子有用以及起点，所以我们用spfa处理出这些格子两两之间的距离（注意细节……这里写挂了好几次），然后设f[s][i]为碰完的机关石状态为s，现在在有用格子的第i个的最小停下次数，转移按照套路即可

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;
const int N=105,dx[]={0,0,1,-1},dy[]={1,-1,0,0};
int n,m,t,a[N][N],f[70005][65],x[N],y[N],bx,by,dis[N][N][5],id[N][5],tot;
char c[N][N];
bool v[N][N];
struct qwe
{
    int x,y;
    qwe(int X=0,int Y=0)
    {
        x=X,y=Y;
    }
};
bool ok(int x,int y)
{
    return x>=1&&x<=n&&y>=1&&y<=m&&c[x][y]!=‘#‘;
}
void spfa(int sx,int sy,int s)
{//cerr<<sx<<" "<<sy<<endl;
    if(!ok(sx,sy))
        return;
    // cerr<<s<<endl;
    queue<qwe>q;
    memset(dis,0x3f,sizeof(dis));
    dis[sx][sy][0]=dis[sx][sy][1]=dis[sx][sy][2]=dis[sx][sy][3]=0;
    v[sx][sy]=1;
    q.push(qwe(sx,sy));
    while(!q.empty())
    {
        int x=q.front().x,y=q.front().y;
        q.pop();
        v[x][y]=0;
        for(int j=0;j<4;j++)
            if(ok(x+dx[j],y+dy[j]))
                for(int k=0;k<4;k++)
                    if(dis[x+dx[j]][y+dy[j]][k]>dis[x][y][j]+(j!=k))
                    {
                        dis[x+dx[j]][y+dy[j]][k]=dis[x][y][j]+(j!=k);
                        if(!v[x+dx[j]][y+dy[j]])
                        {
                            v[x+dx[j]][y+dy[j]]=1;
                            q.push(qwe(x+dx[j],y+dy[j]));
                        }
                    }
    }
    for(int i=1;i<=t;i++)
        for(int j=0;j<4;j++)
            for(int k=0;k<4;k++)
                a[s][id[i][j]]=min(a[s][id[i][j]],dis[x[i]+dx[j]][y[i]+dy[j]][k]+(dx[j]+dx[k]!=0||dy[j]+dy[k]!=0));
}
int main()
{
    scanf("%d%d%d",&n,&m,&t);
    for(int i=1;i<=n;i++)
        scanf("%s",c[i]+1);
    for(int i=1;i<=t;i++)
        scanf("%d%d",&x[i],&y[i]);
    scanf("%d%d",&bx,&by);
    for(int i=1;i<=t;i++)
        for(int j=0;j<4;j++)
            id[i][j]=++tot;
    memset(a,0x3f,sizeof(a));
    for(int i=1;i<=t;i++)
        for(int j=0;j<4;j++)
            spfa(x[i]+dx[j],y[i]+dy[j],id[i][j]);//,cerr<<x[i]<<" "<<dx[j]<<" "<<y[i]<<" "<<dy[j]<<endl;
    spfa(bx,by,tot+1);
    // for(int i=1;i<=tot+1;i++)
    // {
        // for(int j=1;j<=tot+1;j++)
            // cerr<<a[i][j]<<" ";
        // cerr<<endl;
    // }
    memset(f,0x3f,sizeof(f));
    f[0][tot+1]=0;
    for(int i=0,len=(1<<t)-1;i<=len;i++)
        for(int j=1;j<=tot+1;j++)
            for(int k=1;k<=tot;k++)
                f[i|(1<<((k-1)/4))][k]=min(f[i|(1<<((k-1)/4))][k],f[i][j]+a[j][k]+1);
    int ans=1e9;
    for(int i=1;i<=tot;i++)
        ans=min(ans,f[(1<<t)-1][i]);
    printf("%d
",ans);
    return 0;
}