HDU5353.Average(多校第六场)

题意:
有n个人围成一桌,刚开始每个人有a[i]个糖果,每个人能给左边的或者右边的人一个糖果,也可以同时给左右,也可以不给,但是只能给一个
思路:
首先先判断这些糖果能不能平均分配,然后求出每个人一开始的糖果数和平均数的差,
两个人之间只能是你给我或者我给你或者都不给,那么如果确定了1,2两个人之间的关系(可以枚举),那么从第二个人开始,如果第二个人现在的糖果数目小于平均数,那么必定从右边拿一个,如果多的话肯定是给右边的人一个糖果,以此类推,然后循坏一遍之后,如果每个人的糖果数目都等于平均数,输出答案,否则换另外一种1,2之间的关系
时间复杂度:O(3*n)

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define inf -0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a, b) memset(a, b, sizeof(a))
typedef long long ll;
struct node{
    int x,y;
}line[100019];
int a[100019];
int c[100019];
int d[100019];

int main(){
    int t;
    int n;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        ll sum=0;
        int averge;
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            sum+=a[i];
        }
        if(sum%n!=0){
            printf("NO\n");
            continue;
        }
        averge=sum/n;
        for(int i=1;i<=n;i++){
            c[i]=a[i]-averge;
            d[i]=c[i];
        }
        int flag;
        for(int i=0;i<=2;i++){
            for(int i=1;i<=n;i++)
                c[i]=d[i];
            flag=1;
            int cnt=0;
            if(i==0){

            }
            else if(i==1&&a[1]>=1){
                line[++cnt].x=1;
                line[cnt].y=2;
                c[1]--;
                c[2]++;
            }
            else if(i==2&&a[2]>=1){
                line[++cnt].x=2;
                line[cnt].y=1;
                c[1]++;
                c[2]--;
            }
            for(int j=2;j<=n;j++){
                if(c[j]==0)
                    continue;
                else if(c[j]>0){
                    line[++cnt].x=j;
                    line[cnt].y=j+1;
                    if(line[cnt].y>n){
                        line[cnt].y=1;
                    }
                    c[j]--;
                    if(j+1>n)
                        c[1]++;
                    else
                        c[j+1]++;
                }
                else{
                    line[++cnt].y=j;
                    if(j+1>n){
                        line[cnt].x=1;
                    }
                    else{
                        line[cnt].x=j+1;
                    }
                    c[j]++;
                    if(j+1>n)
                        c[1]--;
                    else
                        c[j+1]--;
                }
            }
            for(int i=1;i<=n;i++){
                if(c[i]!=0&&cnt<=n){
                    flag=0;
                    break;
                }
            }
            if(flag==1){
                printf("YES\n%d\n",cnt);
                for(int i=1;i<=cnt;i++){
                    printf("%d %d\n",line[i].x,line[i].y);
                }
                break;
            }
        }
        if(flag==0){
            printf("NO\n");
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值