题目描述
有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
思路:一道思维题,我们需要分三种情况【本题所用图均来自:官方题解】:
我们假设需要分配的房间为room_i,最后一次被分配的房间为room_x,再分配后房间内最少人数为p_min。
(1)room_i>room_x,此时是下图的情况:
按照每轮分配原则,room_x+1和room_i之间被分配p_min个人,room_i向后绕回room_x的房间则被分配pmin+1个人,可以计算出room_i的初始人数为 p_min×n+[n-(room_i-room_x)]。
(2)room_i<room_x,此时:
room_i+1到room_x之间被分配p_min+1个人,其他房间被分配p_min个人。room_i初始人数为 pmin×n+(room_x-room_i)。
(3)room_i=room_x,此时:
这时重合房间被分配0人,初始人数为 p_min×n。
从上述三种情况我们可以发现,无论room_i的位置在哪里,其房间内再分配后的人数一定是最小的,因此我们只需要找到最小人数对应的房间便是room_i的位置,而当有多个最小值出现的时候,则我们从room_x往前退找到第一个出现最小值的位置即为room_i【想想为什么是这样】。随后,根据整数轮次,将每个房间的人数都减去p_min,求出其初始人数。这时只有room_x和room_i重合的情况才会正好分配完,其他情况我们根据最后不足一轮时,两房间之间相隔房间数即可求出room_i初始房间人数。
#include<stdio.h>
#include<limits.h>
#include<algorithm>
using namespace std;
#define ll long long
int n,x;
ll room[1500005];
int main(void){
scanf("%d%d",&n,&x);
ll mn=INT_MAX;
for(int i=0;i<n;i++){
scanf("%lld",&room[i]);
mn=min(mn, room[i]);
}
x-=1;
int pos=x;
while(room[pos]!=mn)
pos=pos>0?pos-1:n-1;
for(int i=0;i<n;i++)
room[i]-=mn;
int count=0;
for(int i=x;i!=pos;i=i>0?i-1:n-1){
room[i]-=1;
count+=1;
}
room[pos]+=count+n*mn;
for(int i=0;i<n;i++)
printf("%lld ",room[i]);
return 0;
}