笔试刷题-头条

题目描述：

/**
有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

例子说明1:
*/

思路如下：

1.最后一个分配的房间里的人数就是最小值，
这种情况这个房间里的人就是先前被请出去了，
也就是循环了theMin圈，很好计算；
2.最后一个分配的房间里的人数不是最小值，
那么先前被请出去的房间就是
从最后一个分配的房间向前数第一个遇到的最小值的房间，
自己画一个图就理解了.

代码如下：

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;

int main()
{
    int n, k;
    while (~scanf("%d%d", &n, &k)) {
        vector<LL> arr;
        LL theMin = 0x3f3f3f3f3f;
        for (int i = 0; i < n; i++) {
            LL x;
            scanf("%lld", &x);
            arr.push_back(x);
            theMin = min(theMin, x);
        }
        if (arr[k - 1] == theMin) {
            for (vector<LL>::size_type i = 0; i < arr.size(); i++) {
                if (i == k - 1)
                    printf("%lld", 1LL * theMin * n);
                else
                    printf("%lld", arr[i] - theMin);
                putchar(i == arr.size() - 1 ? '\n' : ' ');
            }
        } else {
            int i, d;
            for (i = k - 1, d = 0; theMin != arr[(i + n) % n]; i--, d++)
                arr[(i + n) % n] = arr[(i + n) % n] - theMin - 1;
            arr[(i + n) % n] = 1LL * theMin * n + d;
            if ((i + n) % n != k) {
                for (--i; (i + n) % n != k; i--)
                    arr[(i + n) % n] = arr[(i + n) % n] - theMin;
                arr[(i + n) % n] = arr[(i + n) % n] - theMin;
            }
            for (vector<LL>::size_type i = 0; i < arr.size(); i++)
                printf("%lld%c", arr[i], i == arr.size() - 1 ? '\n' : ' ');
        }
    }
    return 0;
}