题目描述:
有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个整数,代表每个房间分配前的人数。
示例:
输入
3 1
6 5 1
输出
4 4 4
解析:
这道题要注意的核心就是他是把某一房间人数排空再排序的,那么,只要选出排序后人数最少的人数,那就是一开始被选中的房间。
但是,这里存在一个陷阱,那就是0人数的房间,当0人数房间数量为1时,表示原房间人数不足一循环;当0人数房间数量大于1时,表示在排序前就存在0人数房间。
因此,总结就是先从末尾的房间开始减人数,直至选中的房间前。这一段是处理循环后的余数;
然后整个数组都减去当初选中的房间的人数就可得答案。
代码:
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int main()
{
int n, x;
vector<int> arr;
cin >> n >> x;
//记录最少人数
int min = 10e9, min_num = 0;
for (int i = 0;i < n;i++)
{
int t;
cin >> t;
if (min >= t)
{
min = t;
min_num = i;
}
arr.push_back(t);
}
//所有房间减去最少人数
for (int i = 0;i < n;i++)
{
arr[i] -= min;
}
//将余数部分清除
int rem = 0;
while (arr[x - 1] != 0)
{
arr[x - 1]--;
x--;
rem++;
if (x == 0)
{
x = n;
}
}
arr[min_num] = n * min + rem;
for (vector<int>::iterator iter = arr.begin();iter != arr.end();iter++)
{
cout << *iter << " ";
}
}