这有一道题!!!选数
首先先放上一道算法题(c++),一道基础题。解法很多,搜索,枚举都可以。今天以枚举来讲解这道题。不再赘述题干。
最容易想到的办法,大概就是两重循环枚举吧。外循环控制执行次数,内循环执行寻找k个数,然后,相加并判断是否为素数。内循环的实现可以用递归或者是搜索。思路很简单,但却很容易超时。那么,思路该是怎样的呢?观察数据量,n=20,k<n。暗示性极强呀,复杂度就是O(2^n)呗。那不就是枚举n的所有子集,选取其中为k个数的子集。
如果,我们将集合中,每个元素出现标记为1,不出现标记为0.那么自然也就是(2n)种情况呀。也就是原问题的每一个子集,对应一个二进制数.最大值是2n-1。同样地也对应于一个10进制数(<2n-1)。这作为我们思路的出发点。
那么,问题自然就是,当给出一个10进制数(<2n-1),当它对应的二进制数中含有k个1时,将这k个1所对应的元素求和,并判断是否为素数即可。思路已经理清。再细化即可得出代码。
(1)枚举0-(2n-1)(此处可以进一步缩小枚举范围,不缩小也不超时)。
(2)对于10进制数,判断对应的二进制的1的个数。一方面可以自己实现,但是c++提供了接口函数__builtin_popcount(),输入是10进制数,返回的是该数对应的二进制数中1的个数(接口的头文件是cstdio或者iostream)。
(3)找二进制中1所对应的元素。对于n个数中的第i个数而言,如果它出现了(它对应的十进制数记为S),那么1<<(i-1)&S==1位true。反之为false。
(这一点建议结合代码理解)。接着,求和
(4)素数判断,略过。
AC源代码,如下:
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=25;
long long cnt;
bool check(int n){
for(int i=2;i*i<n;i++){
if(n%i==0) return false;
}
return true;
}
int main(){
int n,k,a[maxn];
cin>>n>>k;
for(int i=1;i<=n;i++){
cin>>a[i];
}
int U=1<<n;
for(int S=0;S<U-1;S++){
if(__builtin_popcount(S)==k){
int sum=0;
for(int i=1;i<=n;i++){
if((1<<(i-1))&S) sum+=a[i];
}
if(check(sum)) cnt++;
}
}
cout<<cnt;
return 0;
}
还有一道题!!!
这道题也是输出组合数,但是要求的话,按照字典序排序。思路和刚才类似的。同样地,利用二进制去表示每个元素是否出现。但是,无法解决按字典排序。即使,利用排序算法,也很容易超时。
变换一下思路:这次,倒着表示(假设一共三个数)。二进制最高位表示数字1,最低位表示数字3。并且从2n-1枚举到n。并注意输出格式。手模拟一下,这样是不是正好满足条件。
AC代码如下:
#include<iostream>
#include<cstdio>
#include<stack>
using namespace std;
int main(){
int n,r;
cin>>n>>r;
long long U=1<<n;
for(long long i=U-1;i>=0;i--){
if(__builtin_popcount(i)==r){
stack<int> s;
for(int j=1;j<=n;j++){
if((1<<(j-1))&i) s.push((n-j+1));
}
while(!s.empty()){
printf("%3d",s.top());
s.pop();
}
cout<<endl;
}
}
return 0;
}
以上是两道关于组合的问题。虽然,能够解决。但是算法效率太有限了。因为复杂度是O(2n)。所以,不合适大规模问题,仍有待提高。
接着,讲一讲排列的问题(数学原理略过)。求排列可以借助STL库中,函数next_permutation()。
#include<algorithm>
//对于[start,end)。按照字典序,产生下一个全排列
next_permutation(start,end)。
//输入是数组的地址。
//返回值,若仍然有排列,返回true并改变原数组。若已经产生全部则返回0。
简单举个例子:
输入的数组:2,3,1
调用一次后数组:3,1,2
调用两次后数组:3,2,1
再次调用返回0。
还有一道题
AC代码如下:
#include<algorithm>
#include<cstdio>
#include<iostream>
using namespace std;
const int maxn=1e4+5;
int a[maxn],b[maxn];
int N,M;
bool check(int* a,int* b){
for(int i=i;i<=N;i++){
if(a[i]!=b[i]) return false;
}
return true;
}
int main(){
cin>>N>>M;
for(int i=1;i<=N;i++){
cin>>a[i];
}
while(M--) next_permutation(a+1,a+N+1);
for(int i=1;i<=N;i++){
cout<<a[i];
if(i!=N) cout<<" ";
}
return 0;
}