编程题2-房间的人数(字节跳动2018校招)

题目描述

有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

输出： 4 4 4

算法分析

分配后的房间里，人数最少的那一个房间就是i号房间。如果有多个房间人数最少，则取x号房间往前数的第一个人最少的房间。综上：无论哪种情况，从x房间往前数找到的第一个人最少的房间就是i号房间。

找出i号房间后，就很容易求出之前的人数了。首先根据i号房间最后的人数确定完整的轮数，然后从x号往前再减去最后不足一轮的部分，然后把多出的人数全部补到第i个房间，就结束了

提交代码：

#include<iostream>
#include<vector>
#include<algorithm>
#include <iterator>

using namespace std;

int main()
{
	long long n, x;
	cin >> n >> x;
	vector<long long> arr(n + 1, 0);

	for (int i = 1; i <= n; ++i) cin >> arr[i];

	// 房间i的人数最小,如果相同,取倒序离x最近的点
	vector<int> room_i;
	long long min_val = *min_element(arr.begin() + 1, arr.end());

	int room_index = x;
	do {
		if (arr[room_index] == min_val) break;
		--room_index;
		if (room_index < 1) room_index = n;
	} while (room_index != x);

	// 每个数减去轮数min_val
	for (int i = 1; i <= n; ++i) {
		arr[i] -= min_val;
	}
	arr[room_index] = n * min_val;
	// 不足一轮

	while (x != room_index) {
		--arr[x], ++arr[room_index];
		--x;
		if (x < 1) x = n;
	}
	for (int i = 1; i <= n; ++i)
		cout << arr[i] << " ";
	cout << endl;

	return 0;
}