kmjp's blog

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

yukicoder : No.3244 Range Multiple of 8 Query

ちょっと面倒だけどこちらもすんなり解けた。
https://yukicoder.me/problems/no/3244

問題

1~9の数字で構成される文字列Sが与えられる。
クエリとしてSの部分列が指定される。

Sを整数値とみなしたとき、8の倍数となるようにしたい。
隣接する数字を1か所入れ替えることが任意回数行える時、最小何回入れ替えればよいか。

解法

整数が8の倍数となるには、下3桁が8の倍数であればよい。
よって、下3桁を総当たりしよう。そしてその3つの数字を、Sの部分列のうち極力後ろから取ってくるようにすれば入れ替え回数を小さく抑えることができる。

int N,Q;
string S;
vector<int> P[10];

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	FOR(i,10) {
		P[i].push_back(-3);
		P[i].push_back(-2);
		P[i].push_back(-1);
	}
	
	cin>>N>>Q>>S;
	FOR(i,N) {
		P[S[i]-'0'].push_back(i);
	}
	
	vector<vector<int>> cand;
	FOR(i,1000) if(i%8==0) {
		vector<int> V={i%10,i/10%10,i/100%10};
		if(V[0]==0) continue;
		if(V[1]==0) continue;
		if(V[2]==0) continue;
		cand.push_back(V);
	}
	
	while(Q--) {
		int L,R;
		cin>>L>>R;
		L--;
		if(R-L==1) {
			if(S[L]=='8') {
				cout<<0<<endl;
			}
			else {
				cout<<-1<<endl;
			}
		}
		else if(R-L==2) {
			if(((S[L]-'0')*10+(S[L+1]-'0'))%8==0) {
				cout<<0<<endl;
			}
			else if(((S[L]-'0')+(S[L+1]-'0')*10)%8==0) {
				cout<<1<<endl;
			}
			else {
				cout<<-1<<endl;
			}
		}
		else {
			int pos[10];
			int ret=1<<30;
			FOR(i,10) {
				pos[i]=lower_bound(ALL(P[i]),R)-P[i].begin();
			}
			
			FORR(v,cand) {
				int tp[10];
				FOR(i,10) tp[i]=pos[i];
				int p[3];
				p[0]=P[v[0]][--tp[v[0]]];
				p[1]=P[v[1]][--tp[v[1]]];
				p[2]=P[v[2]][--tp[v[2]]];
				if(p[0]<L) continue;
				if(p[1]<L) continue;
				if(p[2]<L) continue;
				int tmp=0;
				tmp=(R-1)-p[0];
				if(p[1]>p[0]) p[1]--;
				if(p[2]>p[0]) p[2]--;
				tmp+=(R-2)-p[1];
				if(p[2]>p[1]) p[2]--;
				tmp+=(R-3)-p[2];
				ret=min(ret,tmp);
				
			}
			
			
			if(ret>=1<<30) ret=-1;
			cout<<ret<<endl;
			
		}
	}
	
	
}

まとめ

なんか8と関係する出来事あったっけ?