k个bad人比第一个good人先出队.cpp

来自「此为北京大学的ACM练习网站上第1012题的源码」· C++ 代码 · 共 27 行

#include<iostream>
using namespace std;
int test(int k,int m,int cur,int l){
	for(;l>k;){
    	int a=(m%l+cur)%l;
    	if(a<k&&l>k)return 0;
    	else if(l==k+1)return 1;
		l--;
		cur=(a-1)%l;
	}
	return 1;
}
int main(){
	int n,i,A[14],j;
	for(i=1;i<14;i++){
		for(j=i+1;;j += (j % i == 0) ? i + 1 : 1){
			if(test(i,j,2*i-1,2*i)==1){A[i]=j;break;}
		}
	}
	while(cin>>n){
		if(n==0)break;
	    cout<<A[n]<<endl;
	}
	return 0;
}