【EOJ3652】乘法还原（二分图）

题意：

思路：Orz Claris

先找出所有平方项，将与有平方项的数有关的数对暂时忽略，剩下的直接连边就是一张二分图，暴力DFS染色

将有平方项的数两边都加一个，再判字典序即可

我不会判字典序……耽误了一个下午

 #include<cstdio>
 #include<cstring>
 #include<string>
 #include<cmath>
 #include<iostream>
 #include<algorithm>
 #include<map>
 #include<set>
 #include<queue>
 #include<vector>
 #include<bitset>
 #include<ctime>
 using namespace std;
 typedef long long ll;
 typedef unsigned int uint;
 typedef unsigned long long ull;
 typedef pair<int,int> PII;
 typedef vector<int> VI;
 #define fi first
 #define se second
 #define MP make_pair
 #define N      410000
 #define M      51
 #define MOD 1000000007
 #define eps 1e-8
 #define pi     acos(-1)
 #define oo     1e9

 int Data[N],a[N],b[N],x[N],y[N],c[N],flag[N];
 vector<int>q[N];

 void prepare(int *x,int n)
 {
     for(int i=;i<=n;i++) Data[i]=x[i];
     sort(Data+,Data+n+);
     int m=unique(Data+,Data+n+)-Data-;
     for(int i=;i<=n;i++) x[i]=lower_bound(Data+,Data+m+,x[i])-Data;
 }

 void dfs(int u)
 {
     if(!flag[u]) flag[u]=;
     for(int i=;i<=(int)q[u].size()-;i++)
     {
         int v=q[u][i];
         if(!flag[v])
         {
             flag[v]=-flag[u];
             dfs(v);
         }
     }
 }

 int main()
 {
     //freopen("D.in","r",stdin);
     //freopen("D.out","w",stdout);
     int n;
     scanf("%d",&n);
     int m=;
     for(int i=;i<=n;i++) scanf("%d%d",&x[i],&y[i]);
     for(int i=;i<=n;i++) a[++m]=x[i];
     for(int i=;i<=n;i++) a[++m]=y[i];
     for(int i=;i<=m;i++) b[i]=a[i];
     prepare(a,m);
     for(int i=;i<=n;i++) x[i]=a[i];
     for(int i=;i<=n;i++) y[i]=a[i+n];
     for(int i=;i<=m;i++) c[a[i]]=b[i];

     m=;
     for(int i=;i<=n*;i++) m=max(m,a[i]);
     memset(a,,sizeof(a));
     for(int i=;i<=n;i++)
      if(x[i]==y[i]) a[x[i]]=;
     for(int i=;i<=m;i++) q[i].clear();
     for(int i=;i<=n;i++)
      if(!a[x[i]]&&!a[y[i]])
      {
           q[x[i]].push_back(y[i]);
           q[y[i]].push_back(x[i]);
      }
     memset(flag,,sizeof(flag));
     for(int i=;i<=m;i++)
      if(!flag[i]&&!a[i]) dfs(i);
     int n1=,n2=;
     memset(x,,sizeof(x));
     memset(y,,sizeof(y));
     for(int i=;i<=m;i++)
     {
         if(flag[i]==) x[++n1]=i;
           if(flag[i]==) y[++n2]=i;
     }
     for(int i=;i<=m;i++)
      if(a[i]==)
      {
          x[++n1]=i;
         y[++n2]=i;
      }
     sort(x+,x+n1+);
     sort(y+,y+n2+);
     if(n1==n2)
     {
         int flag=;
         for(int i=;i<=n1;i++)
         {
             if(x[i]<y[i]) break;
              if(x[i]>y[i]){flag=; break;}
         }
         if(flag)
          for(int i=;i<=n1;i++) swap(x[i],y[i]);

     }
     if(n1>n2)

     {
         for(int i=;i<=n1;i++) swap(x[i],y[i]);
         swap(n1,n2);
     }
     printf("%d %d\n",n1,n2);
     for(int i=;i<=n1;i++) printf("%d ",c[x[i]]);
     printf("\n");
     for(int i=;i<=n2;i++) printf("%d ",c[y[i]]);
      return ;
 }