17 年头条笔试

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
#include <map>
#include <set>
using namespace std;

/*
* 题目描述： 总共有n个房间，现有一房间需要重新安排
* 安排方式如下： 假设i号房间需要重新安排，先让该房间人全部出来，
* 再按照i+1,i+2……依次安排，每次往里安排一人，到n-1号房间后从0重新循环。
* 输入 ：房间数目和最后一个被安排的人被安排的房间 和最后各房间的人数
* 输出 ：每个房间本来的人数
* sample input : 3 1 6 5 1
* sample output : 4 4 4 
* 解题思路：
* 从当前的最后一个房间，按照安排的逆过程减人（last,last-1,last-2,
* 到0 号房间从n-1重新循环），第一个为0的房间就是被安排的房间，
* 此时其他房间人数为本来的人数，减的次数就是被安排房间本来人数
**********
* 为减少循环次数，可以先将所有房间的人数都按照人数最少房间的标准，
* 集体减去一定人数，使得人数最少的房间人数为1
* 最后将这里减去的人数加到被安排的房间里
* ***************
**/

int main(){
	int n = 0,last = 0,sum = 0;
	cin>>n>>last;
	last--;//输入的位置下标从1，开始换到从0开始
	vector<int> room(n);
	int leastpeople = 0x7fffffff;
	for(int i = 0; i < n; i++){
		cin>>room[i];
		if(room[i] < leastpeople){
			leastpeople = room[i];
		}
	}
	//集体减去人数
	for(int i = 0; i < n; ++i)
		room[i] -= (leastpeople-1);
	//逆过程
	int j =0;
	for(j = last; room[j]!=0;){
		sum++;
		room[j--]--;
		if(j < 0)
			j = n-1;
		j %= n;
		//cout<<"j"<<j<<endl;
	}
	// 加上前面集体减去的人数
	room[j] = sum +(leastpeople-1)*n;

	for(int i = 0; i < n; i++){
		cout<<room[i]<<endl;;
	}
}