其他分享
首页 > 其他分享> > Week3:选数问题——dfs遍历

Week3:选数问题——dfs遍历

作者:互联网

A-选数问题

题目内容
给出n个正整数,从中选出K个数,使满足这些数的和等于S。
求出总共有多少种选择方式。

输入格式
输入包含多个测试样例。
第一行给出一个正整数T(<=100)为测试样例的数量。

接下来对于每个测试样例:
第一行包含三个正整数n,K,S;
第二行包含n个需要被选择的正整数。

输出格式
对于每个测试样例,输出一个整数表示答案。

每个测试样例的答案都各单独占一行。

样例

1
10 3 10
1 2 3 4 5 6 7 8 9 10

4

注意事项

解题思路
这题是典型的dfs遍历:
对于每个数来说,都有被选和不被选两种情况,只需要递归遍历所有的点的所有这两种情况即可。但是这样子时间复杂度会达到O(2^n),所以我们还需要进行剪枝操作。

对于当前的遍历状态来说,在出现以下情况时需要剪枝:

示例代码

#include <iostream>
#include <vector>

using namespace std;

int t;
int n, k, s;
int a[16];
int ans;

//注意:这里的sum是倒着写的,即从目标S往下减
void fun(int nowsum,int i,vector<int> &counter)
{
    if (counter.size() == k && nowsum == 0)
    {
        ans++;//合法可能
        return;
    }
    if (counter.size() > k || nowsum < 0) return;//剪枝
  
    if (i >= n) return;

    fun(nowsum, i + 1, counter);//不选

    counter.push_back(a[i]);
    fun(nowsum - a[i], i + 1, counter);//选择
    counter.pop_back();
}


int main()
{
    cin >> t;
    while (t--)
    {
        ans = 0;
        cin >> n >> k >> s;
        int sum = s;
        vector<int>nowCount;
        for (int i = 0; i < n; i++)
        {
            cin >> a[i];
        }

        fun(s, 0, nowCount);
        cout << ans << endl;;
    }
}

标签:遍历,nowsum,int,选数,counter,样例,dfs,fun,Week3
来源: https://blog.csdn.net/qq_44506233/article/details/104701899