爆搜,按照题意来搜。。。
太难写了,当模拟写的,代码分成了很多块,长度比较难看。。。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define fastio ios::sync_with_stdio(false),cin.tie(NULL),cout.tie(NULL)
const int inf = 1e9 + 7;
const ll lnf = 1e18 + 7;
const int maxn = 1e6 + 10;
ll mod = 1e9 + 7;
int n, m, l;//总数,还可以选多少狼,还有多少人可以说谎
int a[105];
int isw[105], wl;//是不是狼,有多少狼说谎
void dfs(int now)
{
//输出答案,有格式要求
if (now == 0)
{
if (!wl || m != 0 || l != 0)return;
vector<int>ans;
for (int i = n; i >= 1; i--)
if (isw[i] == 1)
ans.push_back(i);
for (int i = 0; i < ans.size(); i++)
{
cout << ans[i];
if (i < ans.size() - 1)
cout << " ";
}
exit(0);
}
int to = abs(a[now]);
int has = (isw[now] != 0);//之前是否被指认
for (int i = 1; i >= 0; i--)
{
//对isw[now]赋值
if (!has)//之前没被某人指认
{
if (i == 0)
isw[now] = -1;//不是狼
else if (i == 1)
{
if (!m)continue;//不能选狼了
//是狼
m--;
isw[now] = 1;
}
}
if (l&& a[now] * isw[to] >= 0)//有说谎次数,结果与不说谎相反
{
l--;
if(isw[now]==1)//狼说谎
wl++;
bool ok = 0, mflg = 0;//先判指认的人有没有已知身份,再判能不能是狼
bool flag = 1;
if (isw[to] == 0)
{
ok = 1;
if (a[now] < 0)
isw[to] = -1;
else
{
if (m)
isw[to] = 1, m--, mflg = 1;
else
flag = 0;
}
}
if (flag)//之前有不合法状态,不进行搜索
{
dfs(now - 1);
if (ok)
{
if (mflg)m++;
isw[to] = 0;
}
}
l++;
if(isw[now]==1)
wl--;
}
if (a[now] * isw[to] <= 0)//不说谎并且可以为真,从上面复制下来的
{
bool ok = 0, mflg = 0;//先判指认的人有没有已知身份,再判能不能是狼
bool flag = 1;//状态是否合法
if (isw[to] == 0)//isw[to]没被赋值
{
ok = 1;
if (a[now] > 0)
isw[to] = -1;
else
{
if (m)
isw[to] = 1, m--, mflg = 1;
else
flag = 0;//不能是狼,不合法
}
}
if (flag)//之前有不合法状态,不进行搜索
{
dfs(now - 1);
if (ok)//之前有了就不需要回溯
{
if (mflg)
m++;
isw[to] = 0;
}
}
}
if (has)break;//has会在指认他的人处回溯
//回溯isw[now]
isw[now] = 0;
if (i == 1)
m++;
}
}
int main()
{
fastio;
wl = 0;
cin >> n >> m >> l;
for (int i = 1; i <= n; i++)
{
char c;
cin >> c;
cin >> a[i];
if (c == '-')
a[i] = -a[i];
}
dfs(n);
cout << "No Solution";
return 0;
}