[蓝桥杯 2023 国 B] AB 路线

【问题描述】
  有一个由 N × M 个方格组成的迷宫，每个方格写有一个字母 A 或者 B。小蓝站在迷宫左上角的方格，目标是走到右下角的方格。他每一步可以移动到上下左右相邻的方格去。
  由于特殊的原因，小蓝的路线必须先走 K 个 A 格子、再走 K 个 B 格子、再走 K 个 A 格子、再走 K 个 B 格子……如此反复交替。
  请你计算小蓝最少需要走多少步，才能到达右下角方格？注意路线经过的格子数不必一定是 K 的倍数，即最后一段 A 或 B 的格子可以不满 K 个。起点保证是 A 格子。

例如 K = 3 时，以下 3 种路线是合法的：
AA
AAAB
AAABBBAAABBB
以下 3 种路线不合法：
ABABAB
ABBBAAABBB
AAABBBBBBAAA

【输入格式】
第一行包含三个整数 N、M 和 K。
以下 N 行，每行包含 M 个字符（A 或 B），代表格子类型。

【输出格式】
一个整数，代表最少步数。如果无法到达右下角，输出 −1。

【样例输入】

4 4 2 
AAAB 
ABAB 
BBAB 
BAAA

【样例输出】

8

【样例说明】
每一步方向如下：下右下右上右下下；路线序列：AABBAABBA。

【评测用例规模与约定】
对于 20% 的数据，1 ≤ N, M ≤ 4。
对于另 20% 的数据，K = 1。
对于 100% 的数据，1 ≤ N, M ≤ 1000，1 ≤ K ≤ 10。

【思路】
感觉用广搜做比较合适，探测每一步的可能情况。
排除几种不能走的情况：
1.下标越界
2.已经走了k个格子，但是当前格子还是这个字母
3.还没走到k个格子，但是已经不再是这个字母了
另外考虑到标记数组问题，因为题目中没有说走过的格子不能再走了，所以在二维的基础上再加一维，vis[i][j][k]表示当前(i,j)点作为第k个格子是否已访问。例如：
2 2 4
AA
AB
(1,1)点的A有两种，分别是a[1][1][1]和a[1][1][3]（分别作为第1个A格子和第3个A格子）

【代码】

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll maxn=1e6+10; 
ll cnt=maxn;
char a[1010][1010];
ll vis[1010][1010][10],dis[1010][1010][10];
int dx[4]={-1,1,0,0},dy[4]={0,0,-1,1};
struct node{
	int x,y,step;
};
queue<node>q;
void bfs(int n,int m,int k){
	while(!q.empty()){
		node temp=q.front();
		q.pop();
		for (int i=0;i<4;i++){
			int x=temp.x+dx[i];
			int y=temp.y+dy[i];
			if (x<1||x>n||y<1||y>m){
				continue;
			}
			if (a[x][y]==a[temp.x][temp.y]&&k==temp.step){
				continue;
			}
			if (a[x][y]!=a[temp.x][temp.y]&&k!=temp.step){
				continue;
			}
			int nowstep=temp.step+1;
			if (nowstep==k+1){
				nowstep=1;
			}
			if (vis[x][y][nowstep]==1){
				continue;
			}
			vis[x][y][nowstep]=1;
			dis[x][y][nowstep]=dis[temp.x][temp.y][temp.step]+1;
			q.push(node{x,y,nowstep});
		}
	}
}
int main(){
  ios::sync_with_stdio(false);
  ll n,l2,t,l1,l3,l4,h=0,m,k,cnt1=0,cnt2=0,cnt3=0,flag=0,ma=-1;
  cin>>n>>m>>k;
  for (int i=1;i<=n;i++){
	for (int j=1;j<=m;j++){
		cin>>a[i][j];
	}
  }
  vis[1][1][1]=1;
  q.push(node{1,1,1});
  bfs(n,m,k);
  for (int i=1;i<=k;i++){
	if (dis[n][m][i]!=0) cnt=min(cnt,dis[n][m][i]);
  }
  if (cnt!=maxn) cout<<cnt;
  else cout<<-1;
  return 0;
}