Codeforces ECR 83 C. Adding Powers (位运算)

题意:给你n个数和一个底数k,每个数每次能减去k^i(i=0,1,2,....),每个k^i只能用一次,问是否能够将每个数变为0.

题解:我们将每个数转化为k进制,因为每个k^i只能用一次,所以我们统计每个位置上的cnt,若cnt>1，则需要多次用到k^i,则不满足条件.

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <stack>
#include <queue>
#include <vector>
#include <map>
#include <set>
#include <unordered_set>
#include <unordered_map>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 1e6 + 10;
const int mod = 1e9 + 7;
using namespace std;
typedef pair<int,int> PII;
typedef pair<long,long> PLL;
 
int t;
int n,k;
ll a[N];
int cnt[N];
 
bool check(){
    me(cnt,0,sizeof(cnt));
    for(int i=0;i<n;++i){
        int pos=0;
        while(a[i]){
            cnt[pos]+=a[i]%k;
            a[i]/=k;
            pos++;
        }
    }
    for(int i=0;i<=60;++i){
        if(cnt[i]>1) return false;
    }
    return true;
}
 
int main() {
    ios::sync_with_stdio(false);
    cin>>t;
     while(t--){
         cin>>n>>k;
          for(int i=0;i<n;++i) cin>>a[i];
          if(check()) puts("YES");
          else puts("NO");
     }
 
 
 
    return 0;
}