nswoj 2326 ZY的密室

点击打开链接

搜索，

把地图标记多开一维标记钥匙，

然后结构中加个vector 记录走到该点打了那些怪兽，这样遍历一下就可以了。

#include <bits/stdc++.h>

using namespace std;
const int maxn=111;
int n,key;
char ma[maxn][maxn];
int v[maxn][maxn][11];
int dir[4][2]={1,0,0,1,-1,0,0,-1};
struct node{
    int x,y,k;
    int step;
    vector<int>vec;
    friend bool operator < (node a,node b){
        return a.step>b.step;
    }
}ed,st,u,e;
int judge(node a){
    if(a.x<1||a.x>n||a.y<1||a.y>n) return 0;
    if(ma[a.x][a.y]=='#') return 0;
    return 1;
}
void bfs(){
    priority_queue<node> que;
    memset(v,0,sizeof(v));
    while(!que.empty()) que.pop();
    st.step=0;st.k=0;
    que.push(st);
    v[st.x][st.y][0]=1;
    while(!que.empty()){
        e=que.top();
        que.pop();
        for(int i=0;i<4;++i){
            u.x=e.x+dir[i][0];
            u.y=e.y+dir[i][1];
            u.step=e.step+1;u.k=e.k;
            u.vec=e.vec;
            if(judge(u)&&!v[u.x][u.y][u.k]){
                if(ma[u.x][u.y]>='1'&&ma[u.x][u.y]<='9'){
                    int num=ma[u.x][u.y]-'0';
                    if(num==u.k+1){//get key
                        v[u.x][u.y][num]=1;
                        u.k+=1;
                        que.push(u);
                    }else {//not get the key
                        v[u.x][u.y][u.k]=1;
                        que.push(u);
                    }
                }
                else if(ma[u.x][u.y]=='S'){ // beat s
                    int f=0;
                    for(int j=0;j<u.vec.size();++j){
                        if(u.vec[j]==u.x*100+u.y) f=1;
                    }
                    if(!f) u.step+=1,u.vec.push_back(u.x*100+u.y);
                    v[u.x][u.y][u.k]=1;
                    que.push(u);
                }else{ // go
                    if(ma[u.x][u.y]=='T'&&u.k==key) {
                        printf("%d\n",u.step); return ;
                    }
                    v[u.x][u.y][u.k]=1;
                    que.push(u);
                }
            }
        }
    }
    puts("impossible");
}
int main(){
    while(~scanf("%d %d",&n,&key)){
        for(int i=1;i<=n;++i)
            scanf("%s",ma[i]+1);
        for(int i=1;i<=n;++i){
            for(int j=1;j<=n;++j){
                if(ma[i][j]=='K') st.x=i,st.y=j;
                if(ma[i][j]=='T') ed.x=i,ed.y=j;
            }
        }
        bfs();
    }
    return 0;
}