Adding Powers
Suppose you are performing the following algorithm. There is an array v1,v2,…,vn filled with zeroes at start. The following operation is applied to the array several times — at i-th step (0-indexed) you can:
either choose position pos (1≤pos≤n) and increase vpos by ki;
or not choose any position and skip this step.
You can choose how the algorithm would behave on each step and when to stop it. The question is: can you make array v equal to the given array a (vj=aj for each j) after some step?
Input
The first line contains one integer T (1≤T≤1000) — the number of test cases. Next 2T lines contain test cases — two lines per test case.
The first line of each test case contains two integers n and k (1≤n≤30, 2≤k≤100) — the size of arrays v and a and value k used in the algorithm.
The second line contains n integers a1,a2,…,an (0≤ai≤1016) — the array you’d like to achieve.
Output
For each test case print YES (case insensitive) if you can achieve the array a after some step or NO (case insensitive) otherwise.
Example
inputCopy
5
4 100
0 0 0 0
1 2
1
3 4
1 4 1
3 2
0 1 3
3 9
0 59049 810
outputCopy
YES
YES
NO
NO
YES
Note
In the first test case, you can stop the algorithm before the 0-th step, or don’t choose any position several times and stop the algorithm.
In the second test case, you can add k0 to v1 and stop the algorithm.
In the third test case, you can’t make two 1 in the array v.
In the fifth test case, you can skip 90 and 91, then add 92 and 93 to v3, skip 94 and finally, add 95 to v2.
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 35, M = 60;
ll k, a[N], res[M];
int n, t;
int main() {
scanf("%d", &t);
while (t--) {
scanf("%d%lld", &n, &k);
for (int i = 1; i <= n; i++)
scanf("%lld", &a[i]);
memset(res, 0, sizeof(res));
bool flag = true;
for (int i = 1; i <= n; i++) {
int pos = 0;
while (a[i]) {
int mod = a[i] % k;
if ((mod != 1 && mod != 0) || (mod && res[pos])) {
flag = false;
break;
}
if (mod) res[pos] = mod;
pos++;
a[i] /= k;
}
if (!flag)break;
}
puts(flag ? "YES" : "NO");
}
return 0;
}
推荐阅读
-
[codeforces 1312C] Adding Powers 进制转换
-
C. Adding Powers
-
Adding Powers
-
Adding Powers
-
Adding Unit Tests to an existing iOS project with Xcode 4 博客分类: 手机开发 iosunittest
-
Oracle 11g维护分区(一)Adding Partitions
-
powers.exe是什么进程 powers进程查询
-
Password is required when adding a database to AG group if the database has a master key
-
POJ 1730 Perfect Pth Powers G++ pow未实现 素数+gcd实现 巧妙
-
Additive Powers-of-Two (APoT) Quantization:硬件友好的非均匀量化方法