A. Doremy's Paint 3

Q. -> https://codeforces.com/problemset/problem/1890/A
Ans->
#include <bits/stdc++.h>
using namespace std;
int main() {
    int cases;
    cin>>cases;
    while(cases--){
        int n,x;
        cin>>n;
          vector<int>v;
        map<int,int>mp;
        for( int i=0; i<n ;i++){
            cin>>x;
            mp[x]++;
        }
        if( mp.size()>=3){
            cout<<"NO"<<endl;
           
        }
       else if(mp.size()==1){
            cout<<"YES"<<endl;
        }
        else{
      vector<int>a;
      for( auto i : mp){
          a.push_back(i.second);
      }
      if(abs(a[0]-a[1]) <=1){
          cout<<"yes"<<endl;
      }
      else{
          cout<<"no"<<endl;
      }
        }
        
    }
 
    return 0;
}