题目链接
题意:给定一个元素全为0的数组以及k,和数组大小n。
从i = 0开始操作。每一次你都能在任意元素上加上k^i,不对这个元素进行任何操作,但是每一个下标i只能用一次。最终要把这个数组变成题中的给定数组。如果可能输出yes
思路:进制转换:我们可以发现我们一直在加上k的i次方,你可以联想到2进制数也是这样的表示一个数的。那么我们可以把题中给定的数变成k进制的数,如果这个数是完全是由0或者1组成的并且这个数组中所有的元素的i没有重复使用。
列:810的9进制数为 1 1 0 0,59049的9进制数为 1 0 0 0 0 0,求出其进制数后,用vis数组把为1的地方标记了,避免重复。如果大于1,也不行。
#include <bits/stdc++.h>
using namespace std;
const int N = 1e6;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-10;
const int mod = 1e9 + 7;
const int II = 3.1415926535;
typedef long long ll;
int vis[1000];
bool cmp(ll a,ll k)
{
ll arr[1000] = {0};
int i = 0;
while(a != 0)
{
arr[i++] = a%k;
if(a % k > 1)
return false;
a /= k;
}
for(i--;i >= 0;i--)
{
if(vis[i] && arr[i] == 1)
return false;
else if(arr[i] == 1)
vis[i] = 1;
}
return true;
}
int main()
{
std::ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
int t;
cin >> t;
while(t--)
{
memset(vis,0,sizeof(vis));
ll n,k;
cin >> n >> k;
ll arr[n+5] = {0};
for(int i = 0;i < n;i++)
cin >> arr[i];
bool ans = true;
for(int i = 0;i < n;i++)
{
if(cmp(arr[i],k))
continue;
else
{
ans = false;
break;
}
}
if(ans)
cout << "YES" << endl;
else
cout << "NO" << endl;
}
}