C. Game with Multiset

最新推荐文章于 2024-11-12 12:29:51 发布

KQYaili

最新推荐文章于 2024-11-12 12:29:51 发布

阅读量530

点赞数 5

文章标签： c语言算法 c++

本文链接：https://blog.csdn.net/KQYaili/article/details/135152071

版权

In this problem, you are initially given an empty multiset. You have to process two types of queries:

ADD $x$ — add an element equal to $2^{x}$ to the multiset;
GET $w$ — say whether it is possible to take the sum of some subset of the current multiset and get a value equal to $w$ .

Input

The first line contains one integer $m$ ( $\le m \le 10^5$ ) — the number of queries.

Then $m$ lines follow, each of which contains two integers $t_i$ , $v_i$ , denoting the $i$ -th query. If $t_i = 1$ , then the $i$ -th query is ADD $v_i$ ( $\le v_i \le 29$ ). If $t_i = 2$ , then the $i$ -th query is GET $v_i$ ( $\le v_i \le 10^9$ ).
Output

For each GET query, print YES if it is possible to choose a subset with sum equal to $w$ , or NO if it is impossible.

#include<bits/stdc++.h>
using namespace std;
int f[32],cnt[32];
void slove()
{
	int t;
	cin >> t;
	if (t == 1)
	{
		int v;
		cin >> v;
		cnt[v]++;
	}
	else
	{
		int w;
		cin >> w;
		for (int i = 29; i >= 0; i--)
			w -= f[i] * min(w / f[i], cnt[i]);
		cout << (w ? "NO" : "YES") << endl;
	}
}
int main()
{
	int m;
	cin >> m;
	f[0] = 1;
	for (int i = 1; i <= 29; i++)f[i] = f[i - 1] * 2;
	while (m--)
	{
		slove();
	}
}