数字重复但下标不同(xi)也算不同的组合吗?
查看原帖
数字重复但下标不同(xi)也算不同的组合吗?
1325381
joshamash楼主2024/4/10 15:31
#include <bits/stdc++.h>
using namespace std;
const int N = 20;
int n, k, a[N], ans;

bool isPrime(int x){
    if(x==1||x==4) return false;
    if(x==2||x==3) return true;
    if(x%6!=1&&x%6!=5) return false;
    for(int i=2;i<=x/i;i++){
        if(x%i==0) return false;
    }
    return true;
}

void dfs(int u,int cnt,int sum){
    if(cnt>=k){
        if(isPrime(sum)) ans++;
        return;
    }
    for(int i=u;i<n;i++){
        dfs(i+1,cnt+1,sum+a[i]);
        if(a[i]==a[i+1]) i++;//加上这个WA两个
    }
}

int main(){
    cin >> n >> k;
    for(int i=0;i<n;i++) cin >> a[i];
    sort(a,a+n);
    dfs(0,0,0);
    cout << ans;
    return 0;
}
2024/4/10 15:31
加载中...