kmjp's blog

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

yukicoder : No.2213 Neq Move

これも思いつけば力技で。
https://yukicoder.me/problems/no/2213

問題

2変数X,Yがあり、初期値はA,Bである。
以下の処理を繰り返し、X=C、Y=Dとするには最小何回処理を行う必要があるか。

  • Xをインクリメントする
  • Yをインクリメントする
  • Xを0にする
  • Yを0にする

ただし、途中X=Yとなるタイミングがあってはならない。

解法

値を座標圧縮し、0,A,B,C,Dと±2位の値だけ考えるようにしよう。
あとは両者の値のペアについて、ダイクストラ法でその状態に至る最小処理回数を求めて行けばよい。

int T;
int A,B,C,D;
ll dp[30][30];
void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>T;
	while(T--) {
		cin>>A>>B>>C>>D;
		vector<int> W;
		W.push_back(0);
		W.push_back(1);
		W.push_back(2);
		FOR(i,4) if(A+2-i>=0) W.push_back(A+2-i);
		FOR(i,4) if(B+2-i>=0) W.push_back(B+2-i);
		FOR(i,4) if(C+2-i>=0) W.push_back(C+2-i);
		FOR(i,4) if(D+2-i>=0) W.push_back(D+2-i);
		sort(ALL(W));
		W.erase(unique(ALL(W)),W.end());
		int N=W.size();
		FOR(y,N) FOR(x,N) dp[y][x]=1LL<<60;
		A=lower_bound(ALL(W),A)-W.begin();
		B=lower_bound(ALL(W),B)-W.begin();
		C=lower_bound(ALL(W),C)-W.begin();
		D=lower_bound(ALL(W),D)-W.begin();
		priority_queue<pair<ll,int>> Q;
		dp[A][B]=0;
		Q.push({0,A*100+B});
		while(Q.size()) {
			ll co=-Q.top().first;
			x=Q.top().second/100;
			y=Q.top().second%100;
			Q.pop();
			if(dp[x][y]!=co) continue;
			if(y&&dp[0][y]>co+1) {
				dp[0][y]=co+1;
				Q.push({-dp[0][y],y});
			}
			if(x&&dp[x][0]>co+1) {
				dp[x][0]=co+1;
				Q.push({-dp[x][0],x*100});
			}
			if(x+1<N&&x+1!=y&&dp[x+1][y]>co+W[x+1]-W[x]) {
				dp[x+1][y]=co+W[x+1]-W[x];
				Q.push({-dp[x+1][y],(x+1)*100+y});
			}
			if(y+1<N&&y+1!=x&&dp[x][y+1]>co+W[y+1]-W[y]) {
				dp[x][y+1]=co+W[y+1]-W[y];
				Q.push({-dp[x][y+1],x*100+y+1});
			}
		}
		
		
		cout<<dp[C][D]<<endl;
	}
		
}

まとめ

丁寧に場合分けしてもいいんだろうけど、怖かったので力技。