题目大意:
对于每组测试数据,会有 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;
}
浙公网安备 33010602011771号