tyvj P2070 codevs 1316 [NOIP2012P4]文化之旅

这道题也出在常规赛中。然而那个时候A了，却挂在了tyvj和codevs上，于是就来填坑。。。
其实就是爆搜，然后来个A*算法，估计值可先用Floyd算出来。
然后减去那些无法构成的边，即边mp[i][j]当vis[c[j]][ci]]时不成立。。。
然后狂剪枝。。。
我的想法是学了一个点，就把不行的全mark掉。
返回时在修复回来。。。

#include<cstdio>
#include<iostream>
#include<vector>
#include<cstring>
using namespace std;
#define M 105
int n,K,m,s,t,c[M],vis[M][M],dis[M][M],mp[M][M];
int stk[M],top,mark[M];
int ans=2000000000;
struct G{
    int to,v;
};
vector<G>edge[M];
void Min(int &a,int b){
    if(a==-1||a>b)a=b;
}
void dfs(int x,int d){
    if(x==t){
        Min(ans,d);
        return ;
    }
    if(dis[x][t]==-1)return ;
    if(ans<=dis[x][t]+d)return ;
    int sz=edge[x].size(),tmp=top;
    for(int i=0;i<sz;++i){
        G y=edge[x][i];
        if(mark[c[y.to]])continue;
        for(int j=1;j<=K;++j)
            if(!mark[j]&&vis[j][c[y.to]]){
                mark[j]=1;
                stk[++top]=j;
            }
        dfs(y.to,y.v+d);
        while(tmp!=top)mark[stk[top--]]=0;
    }
}
int main(){
    scanf("%d %d %d %d %d",&n,&K,&m,&s,&t);
    for(int i=1;i<=n;++i)scanf("%d",&c[i]);
    for(int i=1;i<=K;++i)
        for(int j=1;j<=K;++j)scanf("%d",&vis[i][j]);
    for(int i=1;i<=K;++i)vis[i][i]=1;
    memset(dis,-1,sizeof(dis));
    for(int i=1;i<=m;++i){
        int x,y,z;
        scanf("%d %d %d",&x,&y,&z);
        Min(dis[x][y],z);
        dis[y][x]=z;
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;++j){
            if(vis[c[j]][c[i]])dis[i][j]=-1;
            if(vis[c[i]][c[j]])dis[j][i]=-1;
        }
    for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j)mp[i][j]=dis[i][j];
    for(int k=1;k<=n;++k)
        for(int i=1;i<=n;++i)
            for(int j=1;j<=n;++j)
                if(dis[i][k]!=-1&&dis[k][j]!=-1)Min(dis[i][j],dis[i][k]+dis[k][j]);
    for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j)
            if(mp[i][j]!=-1)
                edge[i].push_back((G){j,mp[i][j]});
    for(int i=1;i<=K;++i)if(vis[i][c[s]])mark[i]=1;
    dfs(s,0);
    if(ans==2000000000)ans=-1;
    printf("%d\n",ans);
    return 0;
}