题目大意:
对于每组测试数据,会有 m m m 个操作, x x x 为输入的酒的编号, 1 1 1表示在第 x x x 杯酒上贴上红纸, 2 2 2表示在除了第 x x x 杯酒外的酒杯上贴上红纸。
我们可以用 a i a_i ai 来表示第 i i i 杯酒进行过几次 1 1 1 操作,而 s i s_i si 来表示第 i i i 杯酒进行过几次 2 2 2 操作。 c n t cnt cnt 记录在操作一中出现了多少个不同的 x x x 出现, c c cc cc 记录在操作二中出现了多少个不同的 x x x 出现, a n s ans ans 记录至少次数。
当目前为操作一时,如果 c n t cnt cnt 等于 n n n 或者 s x s_x sx 大于 1 1 1,就把 f l a g flag flag 变成true。
当目前为操作二时,如果 c c cc cc 大于等于 2 2 2 时一定可以,或者 a x a_x ax 大于等于 1 1 1, 就可以把 f l a g flag flag 变成true。
最后true就输出 a n s ans ans,否则输出 − 1 -1 −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;
}