C. Adding Powers

图片说明
图片说明

解法

因为选择是可以任意位置的,所以相当于数组是无序的。然后k的每次方要么不用,要么用一次。
所以将数组里面的每个数转换成k进制,然后用一个cnt[]数组来计算,计算k的每次方出现了多少次。
只有所以cnt[i] >= 0 才是yes

代码

#include <bits/stdc++.h>
#define ios ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define debug_in  freopen("in.txt","r",stdin)
#define debug_out freopen("out.txt","w",stdout);
#define pb push_back
#define all(x) x.begin(),x.end()
#define fs first
#define sc second
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll> pii;
const ll maxn = 1e6+10;
const ll maxM = 1e6+10;
const ll inf = 1e8;
const ll inf2 = 1e17;

template<class T>void read(T &x){
    T s=0,w=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
    x = s*w;
}
template<class H, class... T> void read(H& h, T&... t) {
    read(h);
    read(t...);
}

template <typename ... T>
void DummyWrapper(T... t){}

template <class T>
T unpacker(const T& t){
    cout<<' '<<t;
    return t;
}
template <typename T, typename... Args>
void pt(const T& t, const Args& ... data){
    cout << t;
    DummyWrapper(unpacker(data)...);
    cout << '\n';
}

//--------------------------------------------

int T,N,K;
ll a[maxn];
int cnt[100];
bool solve(){
    for(int i = 0;i<100;i++) cnt[i] = 0;
    for(int i = 1;i<=N;i++){
        ll cur = a[i],p = 0;
        while(cur){
            cnt[p] += cur%K;
            cur/=K;
            p++;
        }
    }
    for(int i = 0;i<100;i++){
        if(cnt[i] > 1) return false;
    }
    return true;
}
int main(){
    // debug_in;

    read(T);
    while(T--){
        read(N,K);
        for(int i = 1;i<=N;i++) read(a[i]);
        if(solve()) puts("YES");
        else puts("NO");
    }


    return 0;
}