1312C - Adding Powers - CodeForces Solution

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define vctr vector<ll>
#define vp vector<pair<ll, ll>>

int main()
{
    ll t;
    cin >> t;
    while(t--)
    {
        ll n, i, j, k;
        cin >> n >> k;

        vctr a(n);
        for(i=0; i<n; i++)
        {
            cin >> a[i];
        }

        ll tracker[64]={0}, flag=1;

        for(i=0; i<n; i++)
        {
            while(a[i]!=0)
            {
                ll value=a[i], cnt=0, mul=1;
                while(value>0)
                {
                    value/=k;
                    cnt++;
                    mul*=k;
                }
                cnt--;
                mul/=k;
                if(cnt<0)break;
                if(cnt>=0 && tracker[cnt]==0)
                {
                    tracker[cnt]++;
                    a[i]-=mul;
                }else{
                    break;
                }
            }
            if(a[i]!=0){
                flag=0;
                cout << "NO" << endl;
                break;
            }
        }
        if(flag==1)cout << "YES" << endl;

    }
}