DFS——几个数字的和=目标

49 篇文章 0 订阅
26 篇文章 0 订阅

Description:

现给定一个含有n个元素的数组A,要求:从这n个数中选择一些数,这些数的和恰好为k

Input:

多组测试数据。第一行为n(1<=n<=20) 第二行为n个整数,每个数的范围为(-10^8≤A[i]≤10^8) 第三行为整数k(-10^8≤k≤10^8).

Output:

如果能够达到目的,输出”Of course,I can!”; 否则输出”Sorry,I can’t!”.

Sample Input:

4

1 2 4 7

13

4

1 2 4 7

15

Sample Output:

Of course,I can!

Sorry,I can't!

代码:

//
//  main.cpp
//  DFS(几个数求和方案)
//
//  Created by showlo on 2018/4/9.
//  Copyright © 2018年 showlo. All rights reserved.
//

#include <stdio.h>
int vis[22];
int N,aim,a[22],flag,sum;
int dfs(int i,int sum){         //dfs函数
    if (i<N&&vis[i]==0) {
        vis[i]=1;               //标记i为已访问
        if (sum+a[i]==aim)      //如果达到目标,标记flag为1,返回
        {
            flag=1;
            return flag;
        }
        else if (sum<aim)       //如果没有达到目标值,继续搜索
        {
            dfs(i+1, sum);      //sum不包括a[i]
            dfs(i+1, sum+a[i]);//sum包括a[i]
        }
        else
            dfs(i+1, sum);      //不包括a[i]
        vis[i]=0;
    }
    return flag;
}
int main() {
    while (scanf("%d",&N)!=EOF) {
        flag=0;
        for (int i=0; i<N; i++)
            scanf("%d",&a[i]);
        scanf("%d",&aim);
        for (int i=0; i<N; i++)
            vis[i]=0;
        dfs(0,0);
        if (flag==0) {
            printf("Sorry,I can't!\n");
        }
        else
            printf("Of course,I can!\n");
    }
    return 0;
}

感觉自己对dfs的应用还是不熟悉,比如这道题就不太清楚为什么既然标记VIS[i]=1了,后面还要还原,测试了一下不还原确实会出错。有大佬明白的,还请指教,thx!


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
好的,我可以为你解答这个问题。这个问题可以使用深度优先搜索(DFS)算法来解决,同时也可以使用打表的方法。 首先,我们可以通过打表的方法,列出n在不同的情况下所需要的平方数个数: | n | 最少平方数个数 | |---|---| | 1 | 1 | | 2 | 2 | | 3 | 3 | | 4 | 1 | | 5 | 2 | | 6 | 3 | | 7 | 4 | | 8 | 2 | | 9 | 1 | | 10 | 2 | | 11 | 3 | | 12 | 3 | | 13 | 2 | | 14 | 3 | | 15 | 4 | | 16 | 1 | 通过观察表格,我们可以发现,对于任意正整数n,它所需要的最少平方数个数,可以通过以下的递推式计算得出: $$ f(n)=\min_{i=1}^{\lfloor\sqrt{n}\rfloor}(f(n-i^2)+1) $$ 其中,$f(n)$表示正整数n所需要的最少平方数个数,$\lfloor\sqrt{n}\rfloor$表示不大于$\sqrt{n}$的最大整数。 接下来,我们可以通过DFS算法来实现这个递推式。 具体来说,我们可以定义一个函数`dfs(n)`,它表示求解正整数n所需要的最少平方数个数。在这个函数中,我们可以首先判断n是否等于1、4或9,因为这三个数是最小的平方数。如果n是其中的一个,那么它所需要的最少平方数个数就是1。否则,我们可以遍历从1到$\lfloor\sqrt{n}\rfloor$的所有整数i,对于每个i,我们可以递归调用`dfs(n-i*i)`,并将它们的最小值加1,就可以得到n所需要的最少平方数个数。最后,我们将结果返回。 具体实现可以参考以下代码: ```python import math def dfs(n): if n == 1 or n == 4 or n == 9: return 1 res = float('inf') for i in range(1, int(math.sqrt(n)) + 1): res = min(res, dfs(n - i * i) + 1) return res n = int(input()) print(dfs(n)) ``` 当然,对于比较大的n,上述代码可能会超时,因此我们可以添加记忆化搜索来优化它。 完整代码如下:

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值