ラウンド83 C CF

あなたは次のアルゴリズムを実行していると仮定します。配列のV1、V2、...、開始時にゼロで埋めVNがあります。以下の操作は、アレイに数回適用される - i番目の段階(0インデックス)することができますで:

いずれかのKIによって位置pos(1≤pos≤n)と増加VPOSを選択してください。
または任意の位置を選択して、このステップをスキップしません。
あなたは、アルゴリズムは、各ステップとするとき、それを止めるために振る舞う方法を選択することができます。質問です:あなたは、配列Vは、いくつかの工程の後に(VJ = AJ各jについて)指定された配列aに等しくすることができますか?

入力
テストケースの数-最初の行は、T(1≤T≤1000)整数一方を含みます。テストケースごとに2つのライン-次2T株は、テストケースを含みます。

アレイvのサイズおよびアルゴリズムで使用され、値K - 各テストケースの最初の行は、二つの整数nとk(1≤n≤30、2≤k≤100)を含みます。

2行目は含まれていnは整数では、A2、...、(0≤ai≤1016)A1 - あなたが達成したい配列を。

出力
各テストケースの印刷YES(ケース小文字を区別しない)の場合、あなたはそれ以外の場合は、いくつかのステップまたはNO(ケース小文字を区別しない)の後に配列aを達成することができます。

入力コピー
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

出力コピー
YES
YES
NO
NO
YES

int t, n,k;
ll a[55];
int index, ji[64];

int main() {
	scanf("%d", &t);
	while (t--) {
		scanf("%d%d", &n,&k);
		for (int i = 0; i < n; i++)
			scanf("%lld", a + i);
		

		ms(ji, 0);
		bool flag = true;
		for (int i = 0; i < n; i++) {
			index = 0;
			if (!flag) break;

			while (a[i]) {
				if ((a[i] - 1) % k == 0 && ji[index] == 0) {
					a[i]--;
					if(a[i]) a[i] /= k;
					ji[index++] = 1;
				}
				else if (a[i] % k == 0) {
					a[i] /= k;
					index++;
				}
				else {
					flag = false;
					break;
				}
			}
		}

		if (flag)
			printf("YES\n");
		else
			printf("NO\n");
	}

	return 0;
}

公開された85元の記事 ウォン称賛7 ビュー2144

おすすめ

転載: blog.csdn.net/Fawkess/article/details/104774617