5777. 【NOIP2008模拟】小x玩游戏

Description

        今天，小x因为太无聊，就在玩游戏。这个游戏有两个队伍，然后他们在游戏里面打来打去。
       但小x遇到了难题。他不知道自己的队友是谁。他只知道总共有两个队伍，每队有n个人和很多组击杀情况。他想问你，现在他能否知道两个队伍分别有谁。你可以帮助小x吗？由于小x是个游戏狂魔，所以他玩了很多局游戏。

 

Input

       第一行有一个t，表示小y共玩了t局游戏。
        接下来有t组数据，每组数据第一行有一个n,m，表示每队有n个人，有m组击杀情况，接下来m行每行两个字符串s1，s2，表示s1杀了s2，其中s1，s2的长度均不超过10（数据保证两个字符串都由小写字母组成）。注意一个人可以被击杀多次、一个人可以被曾经杀死过的人给杀死。
 

Output

​总共有t行，每行一个“YES”或“NO”，YES表示符合题目条件，NO表示不符合。

 

Solution

开双hash来记录字符串，然后连边。

然后判断是否合法。先找到一个点，做如下图的操作：

dfs中若有一个点连另一个点的标记相同，则错了，

假设点3可以到点7，则错了。

最后判断A和B的个数是否等于n即可。

代码

const
  mo1=10000009;
  mo2=10000007;
type
  arr=record
    y,next:longint;
  end;
var
  bo:boolean;
  n,m,nm,mn,tt:longint;
  e:array [0..200001] of arr;
  a,b:array [1..11] of longint;
  ls,v:array [0..4001] of longint;
  ha1,ha2:array [0..mo1,1..2] of longint;

function hash1(x:longint):longint;
var
  i:longint;
begin
  i:=x;
  while (ha1[i,1]<>0) and (ha1[i,1]<>x) do
    i:=(i+1) mod mo1;
  exit(i);
end;

function hash2(x:longint):longint;
var
  i:longint;
begin
  i:=x;
  while (ha2[i,1]<>0) and (ha2[i,1]<>x) do
    i:=(i+1) mod mo2;
  exit(i);
end;

procedure add(x,y:longint);
begin
  inc(mn);
  e[mn].y:=y; e[mn].next:=ls[x];
  ls[x]:=mn;
end;

procedure init;
var
  i,j,l,p,t1,t2,k1,k2,kk1,kk2,x,y:longint;
  s:string;
begin
  fillchar(ha1,sizeof(ha1),0);
  fillchar(ls,sizeof(ls),0);
  fillchar(e,sizeof(e),0);
  readln(n,m);
  nm:=0; mn:=0;
  for i:=1 to m do
    begin
      readln(s);
      l:=length(s); p:=pos(' ',s);
      t1:=0; t2:=0;
      for j:=p-1 downto 1 do
        begin
          t1:=(t1+(ord(s[j])-96)*a[p-j]) mod mo1;
          t2:=(t2+(ord(s[j])-96)*b[p-j]) mod mo2;
        end;
      k1:=hash1(t1); kk1:=hash2(t2);
      if ha1[k1,1]=0 then
        begin
          inc(nm);
          ha1[k1,1]:=t1; ha1[k1,2]:=nm;
          ha2[kk1,1]:=t2; ha2[kk1,2]:=nm;
        end else
        if ha2[kk1,1]=0 then
        begin
          inc(nm);
          ha2[kk1,1]:=t2; ha2[kk1,2]:=nm;
        end;

      t1:=0; t2:=0;
      for j:=l downto p+1 do
        begin
          t1:=(t1+(ord(s[j])-96)*a[l-j+1]) mod mo1;
          t2:=(t2+(ord(s[j])-96)*b[l-j+1]) mod mo2;
        end;
      k2:=hash1(t1); kk2:=hash2(t2);
      if ha1[k2,1]=0 then
        begin
          inc(nm);
          ha1[k2,1]:=t1; ha1[k2,2]:=nm;
          ha2[kk2,1]:=t2; ha2[kk2,2]:=nm;
        end else
        if ha2[kk2,1]=0 then
        begin
          inc(nm);
          ha2[kk2,1]:=t2; ha2[kk2,2]:=nm;
        end;
        add(ha2[kk1,2],ha2[kk2,2]);
        add(ha2[kk2,2],ha2[kk1,2]);
    end;
end;

procedure dfs(x,w:longint);
var
  i:longint;
begin
  if not bo then exit;
  i:=ls[x]; v[x]:=w;
  if w=1 then w:=2
         else w:=1;
  while i<>0 do
    begin
      if v[e[i].y]=0 then dfs(e[i].y,w) else
        if v[e[i].y]<>w then
          begin
            bo:=false;
            exit;
          end;
      i:=e[i].next;
    end;
end;

procedure print;
var
  i,ans:longint;
begin
  ans:=0;
  for i:=1 to nm do
    if v[i]=1 then inc(ans);
  if (ans=n) and (nm-ans=n) and bo then writeln('YES')
                                  else writeln('NO');
end;

procedure try1;
var
  i:longint;
begin
  a[1]:=1; b[1]:=1;
  for i:=2 to 10 do
    begin
      a[i]:=(a[i-1]*107) mod mo1;
      b[i]:=(b[i-1]*97) mod mo2;
    end;
end;

begin
  assign(input,'game.in');
  assign(output,'game.out');
  reset(input);
  rewrite(output);
  try1;
  readln(tt);
  while tt>0 do
    begin
      init;
      fillchar(v,sizeof(v),0);
      bo:=true;
      dfs(1,1);
      print;
      dec(tt);
    end;
  close(input);
  close(output);
end.