下面的问题是在比赛中(现在已经结束)比赛链接。
它似乎是经典硬币面额问题的变体 - 给定一个具有硬币值和数字 n 的数组(k 个元素)。我们需要回答可以计算 n 面额的方法的数量。我们可以解决它,DP
这需要O(n*k)
时间。现在在比赛问题中,不是给出硬币值数组,而是有一个值 m,并且硬币值都是 m ex 的所有可能幂。n= 200, m=3.
所以我们有硬币价值[3^0, 3^1, 3^2, 3^3, 3^4]
,更高的权力对这个例子没有用]。
我DP
在这里使用了方法,但它的给予TLE
。通过查看时间限制testcases<=10000
, n<=10000
, m<=10000
,我假设我们必须在给定n,m
的O(n)
时间内解决它[可能还需要优化这个。请帮我解决这个问题。我的解决方案使用DP
.
#include <bits/stdc++.h>
#include <stdio.h>
using namespace std;
int solve(vector<int>&vec, int n){
cout<<"n= "<<n<<": m= "<<vec.size()<<endl;
int r=n+1, c=vec.size()+1;
vector<int>row(c,0);
vector<vector<int> >dp(r, row);
for(int i=0;i<c;i++)
dp[0][i]=1;
for(int i=1;i<r;i++){
for(int j=1;j<c;j++){
int a=0;
if(i-vec[j-1]>=0)
a=dp[i-vec[j-1]][j];
dp[i][j]=a+dp[i][j-1];
}
}
return dp[n][c-1];
}
int main()
{
ios::sync_with_stdio(false);
int tc,n,m;
cin>>tc;
while(tc--){
cin>>n>>m;
vector<int>temp;
int index=0;
temp.push_back(1);
while(temp[index]*m<=n){
temp.push_back(temp[index]*m);
index++;
}
int result=solve(temp,n);
cout<<result<<endl;
}
return 0;
}