HDU - 5353 多校 A题 Average

题目链接：https://cn.vjudge.net/contest/226118#problem/A

这题一开始就想麻烦了，实际上只需要按照下面的规则来考虑即可：

1. 若糖果不能被平分，则直接输出“NO”
2. 求出平均数，让每个人所拥有的糖果数减去这个平均数。
3. 接下来从第一个人开始，枚举其对第二个人进行的操作，即给第二个人1颗糖，或从第二个人处取一颗糖，或者什么也不做。
4. 这样一来第二个人只能对第三个人操作（因为前面的人已经进行操作了，所以不能再操作前面的人），而且操作是唯一的，操作完成之后必须保证第二个人糖果为0（即手中刚好有平均数颗糖果）， 就这样循环操作下去，若中间出现了除0，1，-1以外的其它值，直接就是“NO”。最后如果能全部置为0，就输出“YES”

代码如下：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <set>
#include <map>
#include <queue>
#include <string>
#include <stack>
using namespace std;

typedef long long LL;
typedef unsigned long long uLL;

const int maxn = 100010;
int a[maxn], b[maxn];
//这里设置枚举的元素的先后顺序是有意义的，否则直接从-1枚举到1的话第二组样例会出问题的。
int o[3] = { 0, -1, 1 };
vector<pair<int, int> > ans;

int main(){
    int T;
    scanf("%d", &T);
    while(T--){
        int n;
        LL sum = 0;
        scanf("%d", &n);
        for(int i = 0; i < n; ++i) {
            scanf("%d", &a[i]);
            sum += a[i];
        }

        if(sum%n != 0) {
            printf("NO\n");
        }
        else {
            LL avg = sum / n;
            for(int i = 0; i < n; ++i)
                a[i] -= avg;
            bool flag = true;
            memcpy(b, a, sizeof(int)*n);
            for(int t = 0; t < 3; ++t) {
                ans.clear();
                memcpy(a, b, sizeof(int)*n);
                flag = true;
                a[0] += o[t];
                a[1] -= o[t];
                if(o[t] == 1) ans.push_back(make_pair(2, 1));
                else if(o[t] == -1) ans.push_back(make_pair(1, 2));
                else ;
                for(int i = 1; i < n; ++i) {
                    if(a[i] == -1) {
                        a[(i+1)%n] -= 1;
                        a[i] += 1;
                        ans.push_back(make_pair((i+1)%n+1, i+1));
                    }else if(a[i] == 0) {
                        continue;
                    }else if(a[i] == 1) {
                        a[(i+1)%n] += 1;
                        a[i] -= 1;
                        ans.push_back(make_pair(i+1, (i+1)%n+1));
                    }else {
                        flag = false;
                        break;
                    }
                }
                if(flag) break;
            }
            if(flag) {
                printf("YES\n%d\n", ans.size());
                for(int i = 0; i < ans.size(); ++i)
                    printf("%d %d\n", ans[i].first, ans[i].second);
            }
            else printf("NO\n");
        }
    }
    return 0;
}