yukicoder : No.1508 Avoid being hit

計算量とか自信をもって行うのがちょっと戸惑うタイプ。
https://yukicoder.me/problems/no/1508

問題

N個のマスが並んでいる。
初期状態で、任意のマスに1つ駒を置けるとする。

その後、以下の処理をQターン繰り返す。

• 駒を1手左右に動かす、またはその場にとどめ動かさない。
• ターン毎に、入力で与えられる2マスのいずれかに駒があってはならない。

条件を満たす駒の配置の一例を求めよ。

解法

駒が存在可能な区間の集合を管理し、愚直にシミュレートしよう。
この集合の数はさほど多くならない。

Qターン過ごして最終的に存在可能な区間が空でないなら、後はシミュレーションしたものを巻き戻す。

int N,Q;
int A[101010],B[101010];

vector<pair<int,int>> R[101010];

int ok(int v,vector<pair<int,int>> R) {
	FORR2(a,b,R) if(a<=v&&v<=b) return 1;
	return 0;
}

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>N>>Q;
	FOR(i,Q) cin>>A[i];
	FOR(i,Q) cin>>B[i];
	R[0]={{1,N}};
	FOR(i,Q) {
		vector<pair<int,int>> V,W;
		
		FORR(p,R[i]) {
			auto r=p;
			r.first=max(1,r.first-1);
			r.second=min(N,r.second+1);
			if(V.size()&&V.back().second+1>=r.first) {
				V.back().second=r.second;
			}
			else V.push_back(r);
		}
		FORR(r,V) {
			if(r.first<=A[i]&&A[i]<=r.second) {
				W.push_back({r.first,A[i]-1});
				W.push_back({A[i]+1,r.second});
			}
			else {
				W.push_back(r);
			}
		}
		V.clear();
		FORR(r,W) {
			if(r.first<=B[i]&&B[i]<=r.second) {
				V.push_back({r.first,B[i]-1});
				V.push_back({B[i]+1,r.second});
			}
			else {
				V.push_back(r);
			}
		}
		FORR(r,V) {
			if(r.first<=r.second) R[i+1].push_back(r);
		}
		
	}
	
	if(R[Q].empty()) {
		cout<<"NO"<<endl;
	}
	else {
		int cur=R[Q][0].first;
		vector<int> ret;
		ret.push_back(cur);
		for(i=Q-1;i>=0;i--) {
			for(x=max(1,cur-1);x<=min(N,cur+1);x++) if(ok(x,R[i])) {
				cur=x;
				break;
			}
			ret.push_back(cur);
		}
		reverse(ALL(ret));
		cout<<"YES"<<endl;
		FORR(r,ret) cout<<r<<endl;
	}
	
}

まとめ

禁止するマス数がもう少し増えても平気そう。面倒だけどね。