字节跳不动笔试题

有n个房间，现在i号房间里的人需要被重新分配，分配的规则是这样的：先让i号房间里的人全都出来，接下来按照 i+1, i+2, i+3, … 的顺序依此往这些房间里放一个人，n号房间的的下一个房间是1号房间，直到所有的人都被重新分配。

现在告诉你分配完后每个房间的人数以及最后一个人被分配的房间号x，你需要求出分配前每个房间的人数。数据保证一定有解，若有多解输出任意一个解。

输入描述:
第一行两个整数n, x (2<=n<=10^5, 1<=x<=n)，代表房间房间数量以及最后一个人被分配的房间号；
第二行n个整数 a_i(0<=a_i<=10^9) ，代表每个房间分配后的人数。

输出描述:
输出n个整数，代表每个房间分配前的人数。

输入例子1:
3 1
6 5 1

输出例子1:
4 4 4

简单解析：找到数列中的最小值min_num，这最小值代表所循环的次数，如1代表又循环到该位一次，2代表循环到该位置两次。此时将所有位置减去这个最小值，剩下的仅仅代表最多循环一次，此时被分配的房间号值为0。设置计数器count，每次加1，从后往前推遇到的第一个0就是该位置，该位置的值为min_num*n+count.

#include <iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main(int argc, char *argv[])
{
	int n, x;
	cin >> n >> x;
	x = x - 1;
	vector<long long > a(n, 0);
	long long  max_num = 9223372036854775807;
	for (int i = 0; i <n; i++)
	{
		cin >> a[i];
		max_num = min(a[i], max_num);
	}
	for (int i = 0; i < n; i++)
	{
		a[i] -= max_num;
	}
	int count = 0;
	while (a[x] != 0)
	{
		a[x]--;
		x = (x + n - 1) % n;//从后往前
		count++;
	}
	a[x] =max_num * n+count;
	for (auto i = a.begin(); i != a.end(); i++)
		cout << *i << " ";
	cout << endl;
	return 0;
}