求助WA!!!
查看原帖
求助WA!!!
688891
封禁用户楼主2022/8/14 20:36
#include<bits/stdc++.h>
using namespace std;
long long a[1000010],cnt0,cnt1,cnt2;
int main(){
    int t,n;
    cin>>t;
    for(int l=1;l<=t;l++){
        cin>>n;
        memset(a,0,sizeof(a));
        for(int i=1;i<=n;i++){
            cin>>a[i];
        }
        if(n==1 || n==2){
            if(a[1]==0 ||a[2]==0){
                cout<<"YES";
                continue;
            }
            else{
                cout<<"NO";
                continue;
            }
        }
        if(n==3){
            sort(a+1,a+n+1);
            a[3]+=a[1]*2;
            a[2]-=a[1];
            a[1]=0;
            if(a[2]%3==0 || a[3]%3==0){
                cout<<"YES";
                continue;
            }
            else{
                if(a[2]%3==a[3]%3 && a[2]%3<=2 &&a[3]%3<=2){
                    cout<<"YES";
                    continue;
                }
                else{
                    cout<<"NO";
                    continue;
                }
            }
        }
        if(n>=4){
            for(int i=1;i<=n;i++){
                if(a[i]==0){
                    cnt0++;
                }
                else{
                    if(a[i]==1){
                        cnt1++;
                    }
                    else{
                        if(a[i]==2){
                            cnt2++;
                        }
                    }
                }
            }
            if(cnt0==n-2 && cnt1==1){
                cout<<"NO";
                continue;
            }
            else{
                cout<<"YES";
                continue;
            }
        }
    }
    return 0;
}
/* by aw_aw*/

dalao们...

2022/8/14 20:36
加载中...