Two Flowers CodeChef - TWOFL

https://vjudge.net/problem/CodeChef-TWOFL

先把颜色相同的合并成一个点，建好图，枚举要取的两种颜色（根据图中所有边决定哪些组合要枚举）即可

错误记录：

1.写了个假的对于诸如1 2 1 2这种数据只能找出3（前3个数）的答案的算法

2.46行写成(i-1)*n+m

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<vector>
 #include<set>
 #include<map>
 using namespace std;
 #define fi first
 #define se second
 #define mp make_pair
 #define pb push_back
 typedef long long ll;
 typedef unsigned long long ull;
 typedef pair<int,int> pi;
 int n,m;
 int a[][],p[][];
 int fa[],d[];
 int sz[],sz2[];
 bool vis[];
 int num[],ta,ans;
 int dx[]={,,,-};
 int dy[]={,-,,};
 int find(int x)    {return x==fa[x]?x:fa[x]=find(fa[x]);}
 vector<int> e[];
 //set<pi> s;
 map<pi,vector<pi>> ma;
 //vector<int> s2[4000100];
 vector<int> t;
 void ins(int a,int b)
 {
     int ta=d[a],tb=d[b];
     if(ta>tb)    swap(ta,tb);
     //s.insert(mp(ta,tb));
     if(a>b)    swap(a,b);
     ma[mp(ta,tb)].pb(mp(a,b));
 }
 int main()
 {
     int i,j,k,fx,fy,x,y;
     scanf("%d%d",&n,&m);
     for(i=;i<=n;i++)
         for(j=;j<=m;j++)
             scanf("%d",&a[i][j]);
     for(i=;i<=n;i++)
         for(j=;j<=m;j++)
             p[i][j]=(i-)*m+j,d[p[i][j]]=a[i][j];
     for(i=;i<=n*m;i++)    fa[i]=i,sz[i]=;
     for(i=;i<=n;i++)
         for(j=;j<=m;j++)
             for(k=;k<;k++)
             {
                 x=i+dx[k];y=j+dy[k];
                 if(x>=&&x<=n&&y>=&&y<=m&&a[i][j]==a[x][y])
                 {
                     fx=find(p[i][j]);fy=find(p[x][y]);
                     if(fx!=fy)    fa[fy]=fx,sz[fx]+=sz[fy];
                 }
             }
     for(i=;i<=n;i++)
         for(j=;j<=m;j++)
         {
             fx=find(p[i][j]);//s2[d[fx]].pb(fx);
             for(k=;k<;k++)
             {
                 x=i+dx[k];y=j+dy[k];
                 if(x>=&&x<=n&&y>=&&y<=m)
                 {
                     fy=find(p[x][y]);
                     if(fx!=fy)    ins(fx,fy);
                 }
             }
         }
     for(i=;i<=n*m;i++)    fa[i]=i,sz2[i]=sz[i],ans=max(ans,sz[i]);
     for(auto &xx:ma)
     {
         t.clear();
         for(auto &yy:xx.se)
         {
             fx=find(yy.fi);fy=find(yy.se);
             if(fx!=fy)    fa[fx]=fy,sz2[fy]+=sz2[fx],ans=max(ans,sz2[fy]),t.pb(fx),t.pb(fy);
         }
         for(auto &tt:t)    fa[tt]=tt,sz2[tt]=sz[tt];
     }
     printf("%d",ans);
     return ;
 }