在思想的天空里自由翱翔。

P9343题解

Posted on 2023-05-16 12:48  rayhjfjwl  阅读(32)  评论(0)    收藏  举报  来源

题目传送门

题目大意:

对于每组测试数据,会有 m 个操作, x 为输入的酒的编号, 1表示在第 x 杯酒上贴上红纸, 2表示在除了第 x 杯酒外的酒杯上贴上红纸。

我们可以用 a[i]​ 来表示第 i 杯酒进行过几次 1 操作,而 s[i]​ 来表示第 i 杯酒进行过几次 2 操作。 cnt 记录在操作一中出现了多少个不同的 x 出现, cc 记录在操作二中出现了多少个不同的 x 出现, ans 记录至少次数。

当目前为操作一时,如果 cnt 等于 n 或者s[x]​ 大于 1,就把 flag 变成true。

当目前为操作二时,如果 cc 大于等于 2 时一定可以,或者 a[x] 大于等于 1, 就可以把 flag 变成true。

最后true就输出 ans,否则输出-1。

AC code:

#include <iostream>
#include <vector>
using namespace std;
int a[200005];//操作一 
int s[200005];//操作二 
int main()
{
	int t;
	cin >> t;
	while(t--)
	{
		int n,m;
		cin >> n >> m;
		for(int i = 1;i <= n;i++)
		{
			a[i] = 0;
			s[i] = 0;
		}
		//初始化 
		int cnt = 0,cc = 0;
		bool flag = false;
		int ans = 0;
		for(int i = 1;i <= m;i++)
		{
			int x,y;
			cin >> x >> y;
			if(flag)
			{
				//因为多组测试数据所以不能break 
				continue;
			}
			if(x == 1)
			{
				a[y]++;
				if(a[y] == 1)
				{
					cnt++;
				}
				if(cnt == n)//操作一中把每杯酒都贴了 
				{
					ans = i;
					flag = true;
				}
				if(s[y])//操作二中有过y 
				{
					ans = i;
					flag = true;
				}
			}
			else
			{
				s[y]++;
				if(s[y] == 1)
				{
					cc++;
				}
				if(cc >= 2)//操作二中至少有两个不同的数 
				{
					ans = i;
					flag = true;
				}
				if(a[y])//y在操作一中出现过 
				{
					ans = i;
					flag = true;
				}
			}
		}
		if(flag)
		{
			cout << ans << endl;
		}
		else
		{
			cout << -1 << endl;
		}
	}
	return 0;
 }