牛客的两道dfs

1.传送门:牛客13594-选择困难症

 

题意:给你k类物品,每类物品有a[i]个每个物品都有一个value,每类物品最多选一个,要求有多少种选法使得总value>m(没要求每类物品都必须选)

题解:很明显是一道dfs的题,但是要剪枝优化,假设我们当前所有物品的总vaule>m，那么我们只要算这件物品之后的组合总数就行了(特别注意每类物品可以不选,即num[0]=0,dfs每类物品的个数要从位置0开始而不是1),用一个mul数组来记录每类的方案数.

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <stack>
#include <queue>
#include <vector>
#include <map>
#include <set>
#include <unordered_set>
#include <unordered_map>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 1e6 + 10;
const int mod = 1e9 + 7;
using namespace std;
typedef pair<int,int> PII;
typedef pair<long,long> PLL;

int k;
ll m,ans,a[20][200],mul[20],num[200];


void dfs(int pos,ll sum){
    if(pos>k) return;
     for(int i=0;i<=num[pos];++i){
         if(a[pos][i]+sum>m){
             ans+=(num[pos]-i+1)*mul[pos+1];
             return;
         }
         dfs(pos+1,sum+a[pos][i]);
     }
}

int main() {
    ios::sync_with_stdio(false);
     while(cin>>k>>m){
         for(int i=1;i<=k;++i){
             cin>>num[i];
             for(int j=1;j<=num[i];++j){
                 cin>>a[i][j];
             }
             sort(a[i]+1,a[i]+1+num[i]);
         }
         ans=0;
         mul[k+1]=1;
         for(int i=k;i>=1;--i) mul[i]=mul[i+1]*(num[i]+1);
         dfs(1,0);
        printf("%lld\n",ans);
     }

    return 0;
}

 

2.传送门:牛客14132-贝伦卡斯泰露 

 

 题意:问是否可以将一个数组变成成两个完全相同的数组(元素必须从子序列选)

题解:可以直接暴搜,a[ne]表示的是我当前要添加给a1或a2的值,假如它等于a1，那么就给a2，否则给a1(即a2放不放数永远看a1)，假如a1或a2的长度>n/2，就说明这种方案不行,回溯.因为无论如何第一个数一定给a1，所以直接dfs(1,0,2);

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <stack>
#include <queue>
#include <vector>
#include <map>
#include <set>
#include <unordered_set>
#include <unordered_map>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 1e6 + 10;
const int mod = 1e9 + 7;
using namespace std;
typedef pair<int,int> PII;
typedef pair<long,long> PLL;

int t;
int n,a[N];
int a1[N],a2[N];

bool dfs(int p1,int p2,int ne){
    if(p1>n/2 || p2>n/2) return false;
    if(ne==n+1) return true;
    if(a[ne]==a1[p2+1]){
        a2[p2+1]=a[ne];
        if(dfs(p1,p2+1,ne+1)) return true;
    }
    a1[p1+1]=a[ne];
    return dfs(p1+1,p2,ne+1);
}

int main() {
    ios::sync_with_stdio(false);
    cin>>t;
     while(t--){
         cin>>n;
          for(int i=1;i<=n;++i) cin>>a[i];

          a1[1]=a[1];
          if(dfs(1,0,2)) printf("Frederica Bernkastel\n");
          else printf("Furude Rika\n");

     }

    return 0;
}