swjtuoj月赛 C题（dfs+剪枝）

最新推荐文章于 2020-10-27 11:31:46 发布

QiufengWang424

最新推荐文章于 2020-10-27 11:31:46 发布

阅读量397

点赞数

分类专栏： ------DFS

本文链接：https://blog.csdn.net/guhaiteng/article/details/63276622

版权

------DFS 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

【cf 487C】【数论+构造】【根据前缀积取模构造序列】

传送门：

http://swjtuoj.handsomehow.com/problem/28/

思路：

暴搜+剪枝即可

通过下列裁剪会加快运算速度

1.将狗粮总数/4，不能整除直接输出no

2.如果有比平均数大的,直接输出no

3.dfs时如果前3个满足,则最后一个一定满足

ps：搜索，dp，数学这些基础内容要加大力度练！！！

代码：

#include <set>
#include <map>
#include <queue>
#include <vector>
#include <math.h>
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
using  namespace  std;

#define ff first
#define ss second
#define pb push_back
#define ll long long
#define mod 1000000007
#define ull unsigned long long
#define mst(ss,b) memset(ss,b,sizeof(ss));
#define pl(x) cout << #x << "= " << x << endl;
const int inf = 0x3f3f3f3f;
const int N = 1e5+5;

int n, a[25], ave;
bool vis[25];

bool dfs(int id, int sum, int div){
    if(sum == ave){
        id = sum = 0;
        div++;
        if(div == 3)return 1;
    }
    for(int i=id; i<=n; i++){
        if(!vis[i]){
            if(sum+a[i] <= ave){
                vis[i] = 1;
                if(dfs(i+1, sum+a[i], div))return 1;
                vis[i] = 0;
            }
        }
    }
    return 0;
}

int  main(){
    int T;
    scanf("%d", &T);
    while(T--){
        mst(vis, 0);
        scanf("%d", &n);
        int mx = 0, sum = 0;
        for(int i=1; i<=n; i++)scanf("%d", &a[i]), sum += a[i], mx = max(mx, a[i]);
        ave = sum/4;
        if(sum % 4 || mx>ave)puts("no");
        else{
            if(dfs(1, 0, 0))puts("yes");
            else puts("no");
        }
    }
    return 0;
}