首先将数据离散化,然后r[x]表示x节点与根节点关系,左闭右开,这样每次输入时y++,求关系时候异或就行。每次合并时让大节点合并到小节点上去。
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
const int nMax=10005;
int p[nMax],r[nMax],x[nMax],y[nMax],ty[nMax],sot[nMax];
int L,M,cnt;
int find(int x)
{
if(p[x]==x) return x;
int par=find(p[x]);
r[x]=(r[x]^r[p[x]]);
return p[x]=par;
}
void merge(int x,int y,int tye)
{
int a=find(x),b=find(y);
if(a<b)
{
p[b]=a; r[b]=(r[x]^r[y]^tye);
}
else
{
p[a]=b; r[a]=(r[x]^r[y]^tye);
}
}
int bi(int l,int r,int a)
{
while(l<=r)
{
int mid=(l+r)>>1;
if(sot[mid]==a) return mid;
else if(sot[mid]>a) r=mid-1;
else l=mid+1;
}
}
void init()
{
for(int i=1;i<=cnt;i++)
p[i]=i,r[i]=0;
}
int main()
{
// freopen("test.txt","r",stdin);
scanf("%d%d",&L,&M);
for(int i=0;i<M;i++)
{
char s[10];
scanf("%d%d%s",&x[i],&y[i],s);
if(s[0]=='e') ty[i]=0;
else ty[i]=1;
if(x[i]>y[i]) swap(x[i],y[i]);
sot[++cnt]=x[i];sot[++cnt]=y[i];
}
sort(sot+1,sot+cnt+1); cnt=0;
for(int i=1;i<=2*M;i++)
if(sot[i]!=sot[i-1])
sot[++cnt]=sot[i];
int ans=-1;
init();
for(int i=0;i<M;i++)
{
int a=bi(1,cnt,x[i]),b=bi(1,cnt,y[i])+1;
int pa=find(a),pb=find(b);
if(pa==pb && (r[a]^r[b])!=ty[i])
{
ans=i; break;
}
else if(pa!=pb) merge(a,b,ty[i]);
}
if(ans==-1) ans=M;
printf("%d\n",ans);
return 0;
}