hdu5353 模拟

如果下标从0到n-1的话,0号和1号之间有三种关系,0给1,1给0,或不操作,0和1之间的关系定了之后,1只能和2号进行操作使得自己达到平均数,然后2号也只能与3号进行一个操作,以此类推,一个O(n)的遍历就可以了

#pragma comment(linker, "/STACK:102400000,102400000")
#include <iostream>
#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
#include<cmath>
#include<map>
#include<set>
#include<queue>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1 | 1
#define lowbit(x) x&(-x)
using namespace std;
const int N=1e5+10;
long long inf=1e15;
const int MOD=1e9+7;
int T,n,m,k,x,y,z,l,tot;
int cas=1;
struct node{
    int x,y;
}ans[N];
int a[N],b[N];
bool all_same(){
    for(int i=0;i<n;i++) if(a[i]!=a[0]) return false;
    return true;
}
bool check(int a[]){ /*判断i从1到n-1,i与i+1操作后是否可行*/
    for(int i=1;i<n;i++){
        if(a[i]==0) continue;
        else if(a[i]==-1){
            ans[tot].x=(i+1)%n;
            ans[tot++].y=i;
        }else if(a[i]==1){
            ans[tot].x=i;
            ans[tot++].y=(i+1)%n;
        }else return false;
        a[(i+1)%n]+=a[i];
    }
    return a[0]==0;
}
bool solve(){
    if(a[0]>=0 && a[0]<=2){ /*这时0号可以给1号*/
        memcpy(b,a,sizeof(a[0])*n);
        tot=0;
        b[0]--;b[1]++;ans[tot].x=0;ans[tot++].y=1;
        if(check(b))   return true;
    }
    if(a[0]<=0 && a[0]>=-2){ /*这时1号可以给0号*/
        memcpy(b,a,sizeof(a[0])*n);
        tot=0;
        b[0]++;b[1]--;ans[tot].x=1;ans[tot++].y=0;
        if(check(b))  return true;
    }
    tot=0;
    return check(a); /*0号与1号无操作*/
}
int main()
{
#ifndef  ONLINE_JUDGE
 freopen("aaa","r",stdin);
#endif
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        long long sum=0;
        for(int i=0;i<n;i++) {
            scanf("%d",a+i);
            sum+=a[i];
        }
        if(sum%n){
            puts("NO");
            continue;
        }
        if(all_same()){
            puts("YES");
            puts("0");
            continue;
        }
        sum/=n;
        for(int i=0;i<n;i++) a[i]-=sum;
        if(solve()) {
            puts("YES");
            printf("%d\n",tot);
            for(int i=0;i<tot;i++) printf("%d %d\n",ans[i].x+1,ans[i].y+1);
        }else puts("NO");
    }
    return 0;
}








评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值