hdu5353(2015多校6)--Average（贪心）

标签：

题目链接：点击打开链接

题目大意：有n个人围城一个环，每一个人手里都有一些糖果，第i个人有ai块。

现在有三种操作：

第i个人给第i+1个人一块。如果i有

第i+1个人给第i个人一块。如果i+1有

什么都不做。

第i个人和第i+1个人之间，可以选择一种操作并执行，问最终能不能让所有人手里的糖相等。

当n = 1 时，永远是YES

当n = 2 时，注意1和2之间只能有一种操作，不存在循环。

当n > 2 时：

糖的总数不能均分到n个人手中，直接NO

可以分开，存在一个循环，那么枚举第1个人对第2个人的操作，那么对于第2个人及以后的人来说它的左侧已经确定了，那么它为了维护自己的糖数是平均值，只能对下一个人进行操作，如果当前比平均值少1，那么从下一个人手里拿一个，如果正好，那么不操作，如果多一个，那么给下面的人一个，其他情况都是无解的。

注意：在操作时，要保证它当前有糖。所以要判断一下。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;
#define LL __int64
int a[100010] ;
int ans[100010] ;
int main() {
    int t , n , i , j , num , k , u , v , flag ;
    LL sum , ave ;
    scanf("%d", &t) ;
    while( t-- ) {
        scanf("%d", &n) ;
        for(i = 0 , sum = 0 ; i < n ; i++) {
            scanf("%d", &a[i]) ;
            sum += a[i] ;
        }
        flag = 1 ;
        ave = sum/n ;
        memset(ans,0,sizeof(ans)) ;
        if( sum%n )
            flag = 0 ;
        else if( n == 2 ) {
            if( abs(a[0]-ave) >= 2 )
                flag = 0 ;
            else if( abs(a[0]-ave) )
                ans[0] = a[0] > a[1] ? 1 : -1 ;
        }
        else if( n > 2 ) {
            for(i = -1 ; i <= 1 ; i++) {
                ans[0] = i ;
                if( ans[0] ) num = 1 ;
                else num = 0 ;
                for(j = 1 ; j < n ; j++) {
                    k = a[j] + ans[j-1] ;
                    if( k == ave )
                        ans[j] = 0 ;
                    else if( k+1 == ave ) {
                        if( a[ (j+1)%n ] ) ans[j] = -1 ;
                        else break ;
                    }
                    else if( k-1 == ave ) {
                        ans[j] = 1 ;
                    }
                    else break ;
                }
                if( j == n && a[0]-ans[0]+ans[n-1] == ave ) break ;
            }
            if( i == 2 ) flag = 0 ;
        }

        if( !flag ){
            printf("NO\n") ;
            continue ;
        }
        for(i = 0 , num = 0 ; i < n ; i++) {
            if( ans[i] ) num++ ;
        }
        printf("YES\n%d\n", num) ;
        for(i = 0 ; i < n ; i++) {
            if( !ans[i] ) continue ;
            u = i+1 ;
            v = (i+1)%n+1 ;
            if( ans[i] == -1 ) swap(u,v) ;
            printf("%d %d\n", u , v ) ;
        }
    }
    return 0 ;
}

hdu5353(2015多校6)--Average（贪心）

标签：

原文地址：http://blog.csdn.net/winddreams/article/details/47340111