寒假算法题练习1

题目:圆桌问题

题目内容
圆桌上围坐着2n个人。其中n个人是好人,另外n个人是坏人。如果从第一个人开始数数,数到第m个人,则立即处死该人;然后从被处死的人之后开始数数,再将数到的第m个人处死……依此方法不断处死围坐在圆桌上的人。试问预先应如何安排这些好人与坏人的座位,能使得在处死n个人之后,圆桌上围坐的剩余的n个人全是好人。
Input
多组数据,每组数据输入:好人和坏人的人数n(<=32767)、步长m(<=32767);
Output
对于每一组数据,输出2n个大写字母,‘G’表示好人,‘B’表示坏人,50个字母为一行,不允许出现空白字符。相邻数据间留有一空行。
Sample Input

2 3
2 4

Sample Output

GBBG

BGGB

解题过程
模拟题,要用vector存所有人的位置,方便去除某个元素,即本题中去除被处死的人,并在每组数据中先初始化每个人的位置

vector<int> pos; //位置

for(int i=1;i<=2*n;i++){
    
     //初始化每个人的位置
	 pos.push_back(i);
} 

每次处死操作都去除一个坏人的位置,用vector的erase()很方便,定义整数k来存储一个坏人的位置并利用公式(k+m-1)%pos.size()不断更新。

int k=0;
for(int i=0;i<n;i++){
    
    
	k = (k+m-1)%pos.size();
	pos.erase(pos.begin()+k);
}

输出的时候是个大坑,考虑了很久还是不断地错,最后发现可能有某组数据总人数正好被五十整除,每五十个会换行,数据末尾也会换行,会多输出一次换行,注意判别条件(i % 50==0 && i != 2*n)。

if(i%50==0 && i!=2*n) cout << endl;

最后,由于是多组数据输入,别忘记每组数据结束时的初始化处理

memset(ans,0,sizeof(ans));
pos.clear();

全部代码:

#include<iostream>
#include<memory.h>
#include<vector>
using namespace std; 
vector<int> pos; //假设全部为好人位置 
char ans[700000];
int main(){
    
    
	int n,m;
	while(cin >> n >> m){
    
    
		
		for(int i=1;i<=2*n;i++){
    
    
			 pos.push_back(i);
		} 
		int k=0;
		for(int i=0;i<n;i++){
    
    
			k = (k+m-1)%pos.size();
			pos.erase(pos.begin()+k);
		}
		for(int i=0;i<pos.size();i++){
    
    
			ans[pos[i]] = 'G';
		}
		
		for(int i=1;i<=2*n;i++){
    
    
			if(ans[i] == 'G') cout << 'G';
			else cout << 'B';
 			if(i%50==0&&i!=2*n) cout << endl;
		}
		cout << endl << endl;
		memset(ans,0,sizeof(ans));
		pos.clear();
	}	
}

猜你喜欢

转载自blog.csdn.net/weixin_45218241/article/details/112385510