C.Adding Powers
题目链接
题目描述:
题目大意:
有T组测试(1≤T≤1000),要求将长度为N(1<N<=30)的全0数组转变成为数组ai(0<=ai<=10e16)
可以对原数组进行j次操作
转化规则有两条:
1.在第j次操作时,可以给原数组v任意位置的值加上k^j;(2<=k<=100)
2.在第j次什么也不做;
所以k^j这个j就只能使用一次。
如果原数组最终能转换到a数组则输出YES,不可以就输出NO。
解题思路:
一个进制转换问题,实质就是将所给的(十进制的)每一个a数组的元素按规则转换成(每一位只能是0或者1的)k进制数。因为每一个kj只能使用一次,也就是说转换后的数,不管是几进制,每一位的值只能是0或者1(>1说明kj用了多次),所以如果转换后的k进制数的某一位>1,则转换失败。
代码:
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<map>
#include<cstring>
using namespace std;
typedef long long ll;
int t;
int n,k;
ll a[35];
map<int,int>mp;
void change(ll x){
int cnt=0;
do{
if(x%k)mp[cnt]+=x%k;
x/=k;
cnt++;
}while(x!=0);
}
int main(){
#ifdef HH
freopen("G:/in.txt","r",stdin);
#endif // HH
scanf("%d",&t);
while(t--){
mp.clear();
memset(a,0,sizeof(a));
scanf("%d%d",&n,&k);
for(int i=0;i<n;i++){
scanf("%lld",&a[i]);
if(a[i])change(a[i]);
}
int flag=1;
for(map<int,int>::iterator it=mp.begin();it!=mp.end();it++){
if(it->second>1)flag=0;
}
if(flag)printf("YES\n");
else printf("NO\n");
}
return 0;
}