yukicoder : No.3199 Key-Door Grid

これはかなりオーソドックス。
https://yukicoder.me/problems/no/3199

問題

H*Wのグリッドが与えられる。
グリッド中には移動不可なセルがあり、またスタート地点・ゴール地点が与えられる。
加えて、一部のセルには鍵が置いてあったり扉が置いてある。
扉のあるセルには、対応する鍵を持っていないと侵入できない。
また、鍵が置いてあるマスを通ると、手持ちの鍵は強制的にそちらに持ち替えられる。

ゴールに至る最短移動回数を求めよ。

解法

現在位置と手持ちの鍵を状態とし、そこに至る最短移動回数をもって、ダイクストラ法を回そう。

int H,W,M;
int dp[505][505][12];
string S[505];
int K[505][505];
int D[505][505];
int SY,SX,GY,GX;
void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>H>>W>>M;
	FOR(y,H) {
		cin>>S[y];
		FOR(x,W) {
			FOR(i,M+1) dp[y][x][i]=1<<30;
			if(S[y][x]=='S') {
				S[y][x]='.';
				SX=x,SY=y;
			}
			if(S[y][x]=='G') {
				S[y][x]='.';
				GX=x,GY=y;
			}
			if(S[y][x]>='1'&&S[y][x]<='9') {
				K[y][x]=S[y][x]-'0';
				S[y][x]='.';
			}
			if(S[y][x]>='a'&&S[y][x]<='i') {
				D[y][x]=S[y][x]-'a'+1;
				S[y][x]='.';
			}
		}
	}
	dp[SY][SX][0]=0;
	queue<int> Q;
	Q.push(SY*100000+SX*10+0);
	while(Q.size()) {
		int cy=Q.front()/100000;
		int cx=Q.front()/10%1000;
		int cur=Q.front()%10;
		Q.pop();
		if(cy==GY&&cx==GX) {
			cout<<dp[cy][cx][cur]<<endl;
			return;
		}
		
		int d[]={0,1,0,-1};
		FOR(i,4) {
			int tx=cx+d[i];
			int ty=cy+d[i^1];
			if(tx<0||tx>=W||ty<0||ty>=H||S[ty][tx]=='#') continue;
			if(D[ty][tx]&&cur!=D[ty][tx]) continue;
			int tcur=cur;
			if(K[ty][tx]) tcur=K[ty][tx];
			if(chmin(dp[ty][tx][tcur],dp[cy][cx][cur]+1)) Q.push({ty*100000+tx*10+tcur});
		}
	}
	cout<<-1<<endl;
		
	
}

まとめ

たまには易しめの★3もいいよね。