欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Adding Powers

程序员文章站 2024-03-17 14:30:46
...

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;
}
相关标签: ACM