n个正整数中找m个,找到其和等于指定的数s的情况数

这是笔试中遇到的题,题目具体是找m个不同的正整数,其和大于s的组合数。

思路:

1.遍历,将所有的数压栈遍历一遍。找到满足的情况,时间复杂度O(C_{n}^{m}),结果超时,方法可以优化,但复杂度依然很高。

#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]来存所有的情况,空间复杂度是\Omega (ms),时间复杂度是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);
}

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值