Help me in solving MEBA problem

help
,
1

My issue

approach bta do bhai log

My code

#include <bits/stdc++.h>
using namespace std;

int main() {
	// your code goes here
	int t;
	cin>>t;
	while(t--)
	{
	    int n;
	    cin>>n;
	    long long int A[n];
	    map<int,int>ans;
	    for(int i=0;i<n;i++)
	    {
	        cin>>A[i];
	        
	        if(A[i] != 0)
	        {
	            ans[A[i]]++;
	        }
	    }
	    
	    if(ans.size()==1 || ans.size() == 0)
	    {
	        cout<<"YES"<<endl;
	    }
	    else
	    {
	        cout<<"NO"<<endl;
	    }
	}

}

Problem Link: Make My Array Equal Practice Coding Problem