題意描述
給定n和k,輸入n個數字,每次可以讓第i個數字減去k的x次冪,每個x只能用一次,問能否將全部數字變爲0
思路
首先預處理k的x次冪,需要注意的是,當出現k的x次冪大於1e16的時候再結束。然後對於每個數字,不斷的減去k的次冪,如果最後不是0,那麼就輸出NO,否則輸出YES。
AC代碼
#include<bits/stdc++.h>
#define x first
#define y second
#define IOS ios::sync_with_stdio(false);cin.tie(0);
using namespace std;
typedef unsigned long long ULL;
typedef pair<int,int> PII;
typedef pair<long,long> PLL;
typedef pair<char,char> PCC;
typedef long long LL;
const int N=2*1e5+10;
const int M=150;
const int INF=0x3f3f3f3f;
const int MOD=998244353;
LL a[M],f[M];
bool st[M];
void solve(){
memset(st,0,sizeof st);
int n,k;cin>>n>>k;
f[0]=1;
int idx=0;
for(int i=1;;i++){
f[i]=f[i-1]*k;
if(f[i]>1e16){
idx=i;
break;
}
}
for(int i=0;i<n;i++) cin>>a[i];
bool zero=true;
for(int i=0;i<n;i++) if(a[i]!=0) zero=false;
if(zero) cout<<"YES"<<endl;
else{
int fg=0;
for(int i=0;i<n;i++){
for(int j=idx;j>=0;j--){
if(a[i]>=f[j]){
if(st[j]){
fg=1;
break;
}else{
a[i]-=f[j];
st[j]=true;
}
}
}
}
for(int i=0;i<n;i++) if(a[i]) fg=1;
if(fg) cout<<"NO"<<endl;
else cout<<"YES"<<endl;
}
}
int main(){
IOS;
int t;cin>>t;
while(t--) solve();
return 0;
}