Adding Powers

Adding Powers

题目链接
题目大意：
给出一组n个元素的整数数组，还有k，问这些元素能否由k^i次方相加而成，每个i只能加一次。
思路：我们可以将每个转化成k进制，然后用桶存，最后判断k进制下每一位是否存在大于1的即可。
代码：

#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <queue>
#include <vector>
using namespace std;
typedef long long ll;
int vis[100];
ll a[35];
int t,n,k;
int main(){
	cin>>t;
	while(t--){
		fill(vis,vis+100,0);
		cin>>n>>k;
		for(int i=1;i<=n;i++) cin>>a[i];
		//sort(a+1,a+1+n);
		bool ff=true;
		int m=-1;
		int p;
		for(int i=1;i<=n;i++){
			p=0;
			while(a[i]){
				vis[p++]+=a[i]%k;
				a[i]/=k;
			}
		}
		for(int i=0;i<64;i++){
			if(vis[i]>1){
				ff=false;
				break;
			}
		}
		if(ff) puts("YES");
		else puts("NO");
	}
	return 0;
}