hdu 5353 Average || 2015 Multi-University Training Contest 6 || 简单题

hdu 5353 Average 请戳

1. 题意：
   有一个大小为n的数组，只对每对相邻的两个元素x，y 进行三种操作之一：
   1）.x给y一个数
   2）.y给x一个数
   3）.不变（不操作）
   结果要让所有的数都相等，输出1，2操作的个数，并且给出具体的操作。

2. 思路：
   就两个点，数组总和是否能够整除n，如果不能就肯定不能满足要求；
   另外一点就是按照题目说的模拟操作。由于任意相邻的元素都只能进行三个操作中的一种，所以可以从左到右边进行遍历，从2开始遍历三次（个人从1开始存数据）。
   第一次，不对第一个元素做任何操作，遍历；
   第二次，第一个元素给第二个元素一个数，遍历；
   第三次，第二个元素给第一个元素一个数，遍历；
   遍历如果存在一个能够满足条件的就按要求输出，不满足。。额。。当然也要输出。
   小弱这里先三次遍历做判断，然后再遍历三遍输出答案。（突然觉得好麻烦），对了还要注意一下特殊测试，n= 2&& abs（a[1] - a[2]) == 4，这个是不满足要求的。

3. 复杂度：
   时间复杂度：O（n）
   空间复杂度：O（n）

4. 代码

/* ***********************************************
Author        :Ilovezilian
Created Time  :2015/8/7 14:40:19
File Name     :1001_2.cpp
 ************************************************ */

#include <bits/stdc++.h>
#define fi(i,n1,n) for(int i = n1; i <= n; i ++)
#define fd(i,n1,n2) for(int i = n1; i >= n2; i --)
#define INF 0x7fffffff
#define ll long long
using namespace std;
const int N = 100100, mod = 1e9+7;
int a[N], b[N], av, ans, n;

bool solve1()
{
    ans = 0;
    fi(i,2,n)
    {
        int r = i + 1;
        if(r > n) r = 1;
        if(a[i] > 0) a[i] --, a[r] ++, ans ++;
        else if(a[i] < 0) a[i] ++, a[r] --, ans ++;
    }

    ///fi(i,1,n) printf("a[%d] = %d\n" ,i, a[i]);
    fi(i,1,n) if(a[i]) return 0;
    return 1;
}

void prt()
{
    fi(i,2,n)
    {
        int r = i + 1;
        if(r > n) r = 1;
        if(b[i] > 0) b[i] --, b[r] ++, printf("%d %d\n", i, r);
        else if(b[i] < 0) b[i] ++, b[r] --, printf("%d %d\n", r, i);
    }
}

bool judge()
{
    if(solve1())
    {
        printf("YES\n%d\n", ans);
        prt();
    //printf("3test\n");
        return 1;
    }

    memcpy(a, b, sizeof(b[0]) * (n + 1));
    if(a[1] >= 0 && a[1] <= 2)
    {
        a[1] --, a[2] ++;
        if(solve1())
        {
            b[1] --, b[2] ++;
            printf("YES\n%d\n", ans+1);
            printf("1 2\n");
            prt();
    //printf("1test\n");
        return 1;
        }
    }

    memcpy(a, b, sizeof(b[0]) * (n + 1));
    if(a[1] <= 0 && a[1] >= -2)
    {
        a[1] ++, a[2] --;
        if(solve1())
        {
            b[1] ++, b[2] --;
            printf("YES\n%d\n", ans+1);
            printf("2 1\n");
            prt();
    //printf("2test\n");
        return 1;
        }
    }

    //printf("4test\n");
    return 0;
}

void solve()
{
    ll sum = 0;
    scanf("%d", &n);
    fi(i,1,n)
    {
        scanf("%d", a + i);
        sum += a[i];
    }

/*个人觉得这个可以不用，是一种剪枝啦！
    bool ok = 1;
    fi(i,1,n) if(a[i] != a[i-1]) { ok = 0; break;}
    if(ok)
    {
        printf("YES\n0\n");
        return;
    }
*/

    if(sum % n || (n == 2 && abs(a[2] - a[1]) == 4))//特判前面那个其实也可以不用的。
    {
        printf("NO\n");
        return;
    }
    av = sum / n;
    fi(i,1,n) b[i] = a[i] = a[i] - av;

    if(!judge()) 
    {
        printf("NO\n");
        return;
    }
}

int main()
{
    int cas;
    scanf("%d", &cas);
    while(cas --) solve();
    return 0;
}