PAT顶级（简单DFS剪枝）——1022 Werewolf (35 分)

1022 Werewolf (35 分)

---

解题思路：

怎么又是搜索啊。。。。。。。。。。。。。给我人做麻了
不过和1021相比这题要简单多了，简单剪枝一下就过了，剪枝稍微想一下都能想出来：

1. 首先是不要无脑枚举每个人是狼、人两个状态一直到最后才去判断狼人==M，在搜索的过程中就可以判断还有没有狼的名额了，如果没有就只能是人，不需要额外做无用功。
2. 题目要求降序输出的序列最大，那我们可以发现倒序搜索按照序号大的人优先搜索他的狼的情况，搜到的第一个答案明显就是降序输出的序列最大的结果，因此只需要倒序搜索，按先狼再人的情况搜，找到一次直接输出答案，并且直接中断之后的搜索

（本来还想尝试做记忆化搜索，因为倒序搜索题目只需要出前n个人都是狼的样例就可以卡住时间，结果发现4ms全过。。。。这河里吗？）

代码：

#include<bits/stdc++.h>
#include<unordered_map>
#include<unordered_set>
#define ll long long
#define pii pair<int,int>
#define pll pair<ll,ll>
#define mp make_pair
#define pb push_back
#define G 6.67430*1e-11
#define  rd read()
#define pi 3.1415926535
using namespace std;
const ll mod = 1e9 + 7;
const int MAXN = 30000005;
const int MAX2 = 300005;
inline ll read() {
    ll x = 0, f = 1;
    char ch = getchar();
    while (ch < '0' || ch>'9') {
        if (ch == '-')
            f = -1;
        ch = getchar();
    }
    while (ch >= '0' && ch <= '9') {
        x = (x << 1) + (x << 3) + (ch ^ 48);
        ch = getchar();
    }
    return x * f;
}
ll fpow(ll a, ll b)
{
    ll ans = 1;
    while (b)
    {
        if (b & 1)ans = ans * a % mod;
        b >>= 1;
        a = a * a % mod;
    }
    return ans;
}
int fa1[1005], fa2[1005];
int find(int* fa, int r) { return fa[r] == r ? r : fa[r] = find(fa, fa[r]); }
void join(int* fa, int x, int y)
{
    int fx = find(fa, x), fy = find(fa, y);
    fa[fx] = fy;
}
ll gcd(ll a, ll b) { return b == 0 ? a : gcd(b, a % b); }
int wolf[105];
int n, m, l;
int vis[105][105];
int s[105];
int ans = 0;
int dfs(int i, int num)
{
    if (ans )
    {
        return 1;
    }
    if (i == -1)
    {
        int c = 0;
        int wc = 0;
        for (int i = 0; i < n; i++)
        {
            if ((s[i] > 0 && wolf[s[i]-1]) || (s[i] < 0 && !wolf[-s[i]-1]))
            {
                c++;
                if (wolf[i])wc++;
            }
        }
        if (c == l && wc >= 1 && wc < m && num == m)
        {
            ans = 1;
            int f = 0;
            for (int i = n - 1; i >= 0; i--)
            {
                if (wolf[i])
                {
                    if(f)cout<<' ';
                    printf("%d", i+1);
                    f++;
                }
            }
        }
        return true;
    }
    if (num < m)
    {
        wolf[i] = 1;
        vis[i][num + 1] = dfs(i - 1, num + 1);
        wolf[i] = 0;
    }
    vis[i][num] = dfs(i - 1, num);
}
signed main()
{
    n = rd, m = rd, l = rd;
    for (int i = 0; i < n; i++)
    {
        s[i] = rd;
    }
    dfs(n - 1, 0);
    if (!ans)cout << "No Solution";
    return 0;
}