这是笔试中遇到的题,题目具体是找m个不同的正整数,其和大于s的组合数。
思路:
1.遍历,将所有的数压栈遍历一遍。找到满足的情况,时间复杂度,结果超时,方法可以优化,但复杂度依然很高。
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main(){
int n,s;
cin>>n>>s;
vector<int>ta(1,1);
int mark=1;
int sum=1;
int re=0;
while(ta.size()>0){
if(ta.size()==n-1){
if(s-sum>ta.back()){
// for(int i=0;i<ta.size();i++)
// cout<<ta[i]<<" ";
// cout<<endl;
re++;
}
mark=0;
}
if(mark){
int temp=ta.back();
ta.push_back(temp+1);
sum+=temp+1;
if(sum>s)
mark=0;
}
else{
if(sum>=s){
sum-=ta.back();
ta.pop_back();
}
else{
ta[ta.size()-1]++;
sum++;
if(sum<s)
mark=1;
}
}
}
cout<<re%1000000007<<endl;
return 0;
}
2.后来网上查了一些类似的题目如:求n个数和为m的组合数目的解法,想到了可以用动态规划来做,有一个矩阵t[m+1][n+1]来存所有的情况,空间复杂度是,时间复杂度是O(nms)。
#include <iostream>
#include <vector>
using namespace std;
int com(vector<int>num,int m,int s){
int n=num.size();
vector<vector<int>>result(m+1,vector<int>(s+1));
vector<vector<int>>temp(m+1,vector<int>(s+1));
for(int i=0;i<n;i++){
if(num[i]>sum)
continue;
else{
for(int j=0;j<m+1;j++){
// temp[j][0]=1;
for(int k=1;k<s+1;k++)
temp[j][k]=result[j][k];
}
temp[0][0]=1;
for(int j=0;j<m;j++){
for(int k=0;k<s+1;k++){
if(k+num[i]<=sum)
result[j+1][k+num[i]]+=temp[j][k];
}
}
}
for(int j=0;j<m+1;j++){
for(int k=0;k<s+1;k++)
cout<<result[j][k]<<" ";
cout<<endl;
}
cout<<"==============="<<endl;
}
}
int main(){
com(vector<int>{1,2,3,4,5,6,7,8,9,10},3,10);
}