牛客练习赛43 F题 Tachibana Kanade Loves Game （容斥）

题目链接：https://ac.nowcoder.com/acm/contest/548/F

题意：玩家初始血量为k, 对手初始血量为p，玩家手中有编号为1~n的n件武器，每使用一件对手就会掉一滴血，每件只能使用一次，对手有编号为2~m的m-1件武器，每件可以无限次重复使用，若玩家使用第i件武器时，对手有一件j武器（满足j是i的因数），则玩家也掉一滴血，谁血量先为0则输掉游戏，问玩家保持最优策略情况下是否能赢得游戏（若玩家初始血量为0则已输）

思路：可以想到，最优策略就是每次都先选择因数不包含2~m的数，可以推断只需要计算2~m中所有质数在1~n中的倍数个数，n减去该个数即为所有可以不伤害自己的攻击选择数，m的范围是2~20，所以最多就8个质数，直接dfs会产生的各个倍数，复杂度为，容斥计算1~n中所有因数包含2~m的数的个数

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <string>
#include <queue>
#include <map>
#include <algorithm>
using namespace std;

typedef long long ll;
const int maxn = 1e5 + 5;
const double eps = 1e-100;
const double pi = acos(-1);

ll k, q, n;
int m;
int p[10] = {2,3,5,7,11,13,17,19 };
ll ans;

void dfs(int f, ll c, int id)
{
	ans += f *(n/c);
	for (int i = id+1; i < 8; i++)
	{
		if (p[i] > m)break;
		dfs(f*(-1), p[i] * c, i);
	}
}

int main()
{
	int t;
	scanf("%d", &t);
	while (t--)
	{
		scanf("%lld%lld%lld%d", &k, &q, &n, &m);
		ans = 0;
		if (n < q || !k)
		{
			printf("QAQ\n");
			continue;
		}
		for (int i = 0; i < 8; i++)
		{
			if (p[i] > m)break;
			dfs(1, p[i], i);
		}
		n -= ans;
		if (q <= n + min(k,ans))
		{
			printf("Yes\n");
		}
		else
		{
			printf("QAQ\n");
		}
	}
}