【NOI2014】挖宝藏(treasure)

Description

Solution

这道题目是WinterCamp 2008 游览计划的加强版，原题是只有一层，但是主要解题思路还是差不多的，好像是一个叫斯坦纳树的东西。解题的关键在于如何把SPFA的单状态转到考虑所有状态，这里就用到了一个自己更新自己，就是枚举一个集合状态，用当前节点的子集状态来预处理这个枚举的子集状态，即：设一个枚举的集合$S$，再枚举$S$的子集$S'$，然后用$f_{x,y,S'}$来更新$f_{x,y,S}$，这样之后再做SPFA时，转移到一个新的节点时，就可以视为有多条路径汇合到这个点上，就能考虑所有情况。

对于不同层的处理

我们假设已经做完了第$i$层，现在去做第$i+1$层，想要往下走的条件是第 i <script type="math/tex" id="MathJax-Element-8">i</script>层的所有宝藏已经全部选完。为了确保这个条件满足，我们引进一个“超级宝藏”，表示上一层的所有点。然后考虑，既然所有点都已经被选完了，那么他们的代价也应该累加上，所以对应点（（x，y）相等）应该继承上一层的贡献。预处理完成就可以开始SPFA了。

Code

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;
#define fo(i,a,b) for(i=a;i<=b;i++)
typedef long long ll;
struct arr{
    int x,y;
}d[1000005];
const int inf=0x7fffffff;
int F[4][2]={-1,0,1,0,0,-1,0,1};
ll f[15][15][1030],g[15][15][1030],c[15][15][15],s[15],p[15][15][15],bz[15][15];
ll h,n,m,i,j,k,x,y,ans;
void dp(int fl){
    int i,j,S,k,l,r,x,y,xx,yy;
    memcpy(g,f,sizeof(f)),memset(f,127,sizeof(f)),memset(bz,false,sizeof(bz));
    fo(i,1,n) fo(j,1,m){
        f[i][j][1]=g[i][j][(1<<(s[fl-1]+1))-1]+c[fl][i][j];
        if(p[fl][i][j]) f[i][j][1<<p[fl][i][j]]=c[fl][i][j];
    }
    fo(S,1,(1<<(s[fl]+1))-1){
        l=r=0;
        fo(i,1,n) fo(j,1,m){
            for(k=(S&(S-1));k;k=((k-1)&S)) f[i][j][S]=min(f[i][j][S],f[i][j][k]+f[i][j][S^k]-c[fl][i][j]);
            d[++r].x=i,d[r].y=j,bz[i][j]=true;
        }
        while(l<r){
            l++;x=d[l].x,y=d[l].y;
            fo(k,0,3){
                xx=x+F[k][0],yy=y+F[k][1];
                if(xx<1||xx>n||yy<1||yy>m) continue;
                if(f[xx][yy][S]>f[x][y][S]+c[fl][xx][yy]){
                    f[xx][yy][S]=f[x][y][S]+c[fl][xx][yy];
                    if(!bz[xx][yy]) bz[xx][yy]=true,d[++r].x=xx,d[r].y=yy;
                }
            }
            bz[x][y]=false;
        }
    }
}
int main(){
    freopen("treasure.in","r",stdin);
    freopen("treasure.out","w",stdout);

    scanf("%d%d%d",&h,&n,&m);
    fo(k,1,h) fo(i,1,n) fo(j,1,m) scanf("%d",&c[k][i][j]);
    fo(i,1,h){
        scanf("%d",&s[i]);
        fo(j,1,s[i]) scanf("%d%d",&x,&y),p[i][x][y]=j;
    }
    fo(k,1,h) dp(k);
    ans=inf;
    fo(i,1,n) fo(j,1,m) ans=min(ans,f[i][j][(1<<(s[h]+1))-1]);
    printf("%lld",ans);
}