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; }