kmjp's blog

競技プログラミング参加記です

yukicoder : No.2366 登校

これはすんなり解けた。
https://yukicoder.me/problems/no/2366

問題

H*Wのグリッドがある。
左上マスから右下マスに移動したい。
その際、隣接マスへの移動は1秒かかる。

一部のマスcは特別であり、コストD(c)を払いC(c)秒時間を巻き戻せる、というパラメータが与えられる。
T秒以内に右下マスに移動するとき、コストの総和の最小値を求めよ。

解法

特別なマスを使わなくても(H+W-1)秒かければ右下マスに到達できる。
よって考えるべき時間帯は-(H+W-1)~(H+W-1)程度の範囲で良い。
あとはマスの位置と時間を状態とし、ダイクストラ法を回せばよい。

int H,W,K,T;
int C[88][88];
int D[88][88];

ll dp[88][88][333];

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>H>>W>>K>>T;
	FOR(i,K) {
		cin>>y>>x>>j>>k;
		y--,x--;
		C[y][x]=j;
		D[y][x]=k;
	}
	FOR(y,H) FOR(x,W) FOR(i,333) dp[y][x][i]=1LL<<60;
	
	priority_queue<pair<ll,ll>> Q;
	dp[0][0][160]=0;
	Q.push({0,900-160});
	
	while(Q.size()) {
		ll co=-Q.top().first;
		int cy=Q.top().second/100000;
		int cx=Q.top().second/1000%100;
		int ct=900-Q.top().second%1000;
		Q.pop();
		if(dp[cy][cx][ct]!=co) continue;
		if(cy==H-1&&cx==W-1&&ct-160<=T) {
			cout<<co<<endl;
			return;
		}
		if(C[cy][cx]) {
			int nt=max(0,ct-C[cy][cx]+1);
			if(chmin(dp[cy][cx][nt],co+D[cy][cx])) {
				Q.push({-dp[cy][cx][nt],cy*100000+cx*1000+900-nt});
			}
		}
		int d[4]={1,0,-1,0};
		FOR(i,4) {
			int ty=cy+d[i];
			int tx=cx+d[i^1];
			int nt=ct+1;
			if(nt>320||ty<0||ty>=H||tx<0||tx>=W) continue;
			if(chmin(dp[ty][tx][nt],co)) {
				Q.push({-dp[ty][tx][nt],ty*100000+tx*1000+900-nt});
			}
		}
		
	}
	cout<<-1<<endl;
	
}

まとめ

Cが大きくて一瞬びっくりした。