房间人数

题目描述：

有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

思路:

需要考虑的情况是

while(room[pos]!=mn)
        pos=pos>0?pos-1:n-1;

最后一个房间的人的前一个最小数才是真正的最小数

代码:

#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;
}

代码效率：

执行用时：0 ms, 在所有 Go 提交中击败了100.00%的用户
内存消耗：2.2 MB, 在所有 Go 提交中击败了100.00%的用户