今日复习了一道全排列型递归枚举
可以用memset(used,false,sizeof used)来批量设置bool数组为false
P9241 [2023 省 B] 飞机降落
本题要注意枚举每个飞机降落时间时,计算当前飞机降落时间可以
如果s<=t[i]+l[i]说明当前飞机能降落
用max(s,t[i])+d[i]来计算飞机的降落时间,其他就是排列型枚举即可
#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
const int N=11;
int t[N],d[N],l[N];
int n;
bool used[N],flag;
void dfs(int u,int s){
if(u>=n){
flag=true;
return;
}
for(int i=1;i<=n;i++){//全排列
if(!used[i]&&s<=t[i]+d[i]){
used[i]=true;
dfs(u+1,max(s,t[i])+l[i]);//更新降落时间
used[i]=false;
}
}
}
int main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int T;cin>>T;
while(T--){
flag=false;
cin>>n;
for(int i=1;i<=n;i++)
cin>>t[i]>>d[i]>>l[i];
memset(used,false,sizeof used);//设置bool数组为false
dfs(0,0);//第几架,上一架飞机降落的时间
if(flag) cout<<"YES"<<endl;
else cout<<"NO"<<endl;
}
return 0;
}
P1036 [NOIP2002 普及组] 选数
本题为组合型枚举,只需按顺序从n个数往后枚举k个最后把sum为素数的答案统计下来输出即可,可以多加一个参数记录已经选了几个数,然后将按顺序往后少于k个的递归进行剪枝
#include <bits/stdc++.h>
using namespace std;
const int N=21;
int a[N],n,k,ans;
bool is_prime(int n){
if(n<=1) return false;
for(int i=2;i<=n/i;i++)
if(n%i==0) return false;
return true;
}
void dfs(int u,int st,int sum){
if(u>=k){
if(is_prime(sum)) ans++;
return;
}
for(int i=st;i<=n;i++){
dfs(u+1,i+1,sum+a[i]);
}
}
int main(){
cin>>n>>k;
for(int i=1;i<=n;i++) cin>>a[i];
dfs(0,1,0);//选了几个,从第几个开始选,和为多少
cout<<ans;
return 0;
}