openjudge 4115 bfs（最短路径+可破墙）

三维数组：vis[i][j][k]第（i，j）个点在剩余为k时是否访问过
http://bailian.openjudge.cn/practice/4115/

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>

using namespace std;
char a[205][205];
int n,m,k,x1,y1,x2,y2,vis[205][205][15],flag=0;
int b[4][2]={{-1,0},{0,1},{1,0},{0,-1}};

struct node
{
    int x,y,k,step;
};

int bfs()//最短路径
{
    queue<node> q;
    q.push({x1,y1,k,0});
    vis[x1][y1][k]=1;
    while(!q.empty())
    {
        node t=q.front();
        q.pop();
        if(t.x==x2&&t.y==y2)
        {
            return t.step;
        }
        for(int i=0;i<4;i++)
        {
            int x=t.x+b[i][0],y=t.y+b[i][1];
            if(x<0||x>=n||y<0||y>=m)
                continue;
            if(vis[x][y][t.k])
                continue;
            if(a[x][y]=='#')
            {
                if(t.k<=0||vis[x][y][t.k-1])
                    continue;
                vis[x][y][t.k-1]=1;
                q.push({x,y,t.k-1,t.step+1});
            }
            else
            {
                vis[x][y][t.k]=1;
                q.push({x,y,t.k,t.step+1});
            }
        }
    }
    return -1;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    memset(vis,0,sizeof(vis));
    cin >> n >> m >> k;
    for(int i=0;i<n;i++)
    {
        cin >> a[i];
        for(int j=0;j<m;j++)
        {
            if(a[i][j]=='@')
            {
                x1=i;
                y1=j;
            }
            else if(a[i][j]=='+')
            {
                x2=i;
                y2=j;
            }
        }
    }
    printf("%d\n",bfs());
    return 0;
}