计蒜客（dfs 等边三角形）

最新推荐文章于 2024-10-31 16:16:13 发布

L_halley

最新推荐文章于 2024-10-31 16:16:13 发布

阅读量254

点赞数 1

文章标签：深度优先算法 c语言

本文链接：https://blog.csdn.net/L_halley/article/details/123156713

版权

这段代码展示了如何用C++实现深度优先搜索（DFS）算法，解决将一组整数和分成三组，使得每组和相等的问题。输入包含整数的个数n和它们的值，首先检查总和是否能被3整除，然后运用DFS进行递归搜索。如果找到解决方案，则输出'yes'，否则输出'no'。

摘要由CSDN通过智能技术生成

代码

#include<bits/stdc++.h>
using namespace std;
int p[25];
int vis[25] = {0};
int n, sum;
bool f = false;
void dfs(int x, int s, int st){
    if(f){
        return;
    }
    if(x == 3){
        f = true;
        return;
    }
    if(s == sum / 3){
        dfs(x + 1, 0, 0);
        return;
    }
    for(int i = st; i < n; i++){
        if(!vis[i]){
            vis[i] = 1;
            dfs(x, s + p[i], i + 1);
            vis[i] = 0;
        }
    }
}

int main()
{
    scanf("%d", &n);
    for(int i = 0; i < n; i++){
        scanf("%d", &p[i]);
        sum += p[i];
    }
    if(sum % 3 != 0){
        printf("no\n");
    }else{
        dfs(0, 0, 0);
        if(f){
            printf("yes\n");
        }else{
            printf("no\n");
        }
    }
    return 0;
}