题目:
若对于整数N,在集合{1,2……,N}中找出m个数,使其和等于剩下的N-m个数的和。返回所有可能的组合数,N<10000(不能使用任何库函数)。
思路:
组合问题都可以通过回溯法+剪枝来解决
根据题目的意思,其实就是将1-N分为和相等的两半,其满足的前提条件是1-N的总和为偶数(只有偶数才能分成相等的两份)。
在满足总和为偶数的情况下,就通过回溯法来求组合的可能性,适当的剪枝可以减小复杂度。
代码:
#include<iostream> #include<vector> using namespace std; #define N 8 int count=0; void Sum(int *nums,vector<int> out,int sum,int total,int n,int idx){ if(idx>n || (sum*2)>total) return; if(sum*2==total){ for(vector<int>::iterator it=out.begin();it!=out.end();it++) cout<<*it<<" "; cout<<endl; count++; return; } out.push_back(nums[idx]); Sum(nums,out,sum+nums[idx],total,n,idx+1); out.pop_back(); Sum(nums,out,sum,total,n,idx+1); } int main(){ int a[N]; for(int i=0;i<N;i++){ a[i]=i+1; } vector<int> out; int total=(1+N)*N/2; if(total%2==0){ Sum(a,out,0,total,N,0); cout<<"Total Count is "<<count<<endl; } else cout<<"Not Found"<<endl; return 0; }