Educational Codeforces Round 83 (Rated for Div. 2) C题

题目链接
题意:给定一个元素全为0的数组以及k，和数组大小n。
从i = 0开始操作。每一次你都能在任意元素上加上k^i，不对这个元素进行任何操作，但是每一个下标i只能用一次。最终要把这个数组变成题中的给定数组。如果可能输出yes

思路:进制转换:我们可以发现我们一直在加上k的i次方，你可以联想到2进制数也是这样的表示一个数的。那么我们可以把题中给定的数变成k进制的数，如果这个数是完全是由0或者1组成的并且这个数组中所有的元素的i没有重复使用。
列:810的9进制数为 1 1 0 0,59049的9进制数为 1 0 0 0 0 0，求出其进制数后，用vis数组把为1的地方标记了，避免重复。如果大于1，也不行。

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-10;
const int mod = 1e9 + 7;
const int II = 3.1415926535;
typedef long long ll;
int vis[1000];
bool cmp(ll a,ll k)
{
    ll arr[1000] = {0};
    int i = 0;
    while(a != 0)
    {
        arr[i++] = a%k;
        if(a % k > 1)
            return false;
        a /= k;
    }
    for(i--;i >= 0;i--)
    {
        if(vis[i] && arr[i] == 1)
            return false;
        else if(arr[i] == 1)
            vis[i] = 1;
    }
    return true;
}
int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    int t;
    cin >> t;
    while(t--)
    {
        memset(vis,0,sizeof(vis));
        ll n,k;
        cin >> n >> k;
        ll arr[n+5] = {0};
        for(int i = 0;i < n;i++)
            cin >> arr[i];
        bool ans = true;
        for(int i = 0;i < n;i++)
        {
            if(cmp(arr[i],k))
                continue;
            else
            {
                ans = false;
                break;
            }
        }
        if(ans)
            cout << "YES" << endl;
        else
            cout << "NO" << endl;
    }
}