牛客假日团队赛6---新游戏

题目链接
https://ac.nowcoder.com/acm/contest/993/I
代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=250+5;
typedef long long ll;
struct node
{
    int v,next;
    bool flag;
}p[maxn*maxn<<2];
typedef pair<int,int>pii;
int tot,cot,mp[maxn][maxn],flag[maxn][maxn],n,num[maxn*maxn],pre[maxn*maxn],head[maxn*maxn];
bool vis[maxn][maxn],tag[maxn*maxn];
set<pii>s;
const int dx[]={0,0,-1,1};
const int dy[]={1,-1,0,0};
void addEdge(int u,int v)
{
    p[cot].v=v;
    p[cot].next=head[u];
    p[cot].flag=1;
    head[u]=cot++;
}
int dfs(int x,int y)
{
    int ans=1;
    vis[x][y]=1;
    flag[x][y]=tot;
    for(int i=0;i<4;i++)
    {
        int nx=x+dx[i],ny=y+dy[i];
        if(nx>=0&&nx<n&&ny>=0&&ny<n&&!vis[nx][ny]&&mp[nx][ny]==mp[x][y])
        {
            ans+=dfs(nx,ny);
        }
    }
    return ans;
}
queue<int>q;
int dfs(int u,int v1,int v2)
{
    int ans=num[u];
    tag[u]=1;
    q.push(u);
    for(int i=head[u];~i;i=p[i].next)
    {
        int v=p[i].v;
        if(!tag[v]&&p[i].flag&&(pre[v]==v1||pre[v]==v2))
        {
            p[i].flag=0;
            ans+=dfs(v,v1,v2);
        }
    }
    return ans;
}
int solve(int u)
{
    int ans=0;
    for(int i=head[u];~i;i=p[i].next)
    {
        int v=p[i].v;
        if(!tag[v]&&p[i].flag)
        {
            ans=max(ans,dfs(u,pre[u],pre[v]));
            while(!q.empty()){tag[q.front()]=0;q.pop();}
        }
    }
    tag[u]=1;
    return ans;
}
int main()
{
    memset(head,-1,sizeof(head));
    int ans=0;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            scanf("%d",&mp[i][j]);
        }
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<maxn;j++)
        {
            if(!vis[i][j])
            {
                ++tot;
                num[tot]=dfs(i,j);
                ans=max(ans,num[tot]);
                pre[tot]=mp[i][j];
            }
        }
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            for(int k=0;k<4;k++)
            {
                int nx=i+dx[k],ny=j+dy[k];
                if(nx>=0&&nx<n&&ny>=0&&ny<n&&flag[nx][ny]!=flag[i][j])
                {
                    int u=min(flag[nx][ny],flag[i][j]),v=max(flag[nx][ny],flag[i][j]);
                    s.insert(pii(u,v));
                }
            }
        }
    }
    set<pii>::iterator it=s.begin();
    for(;it!=s.end();it++)
    {
        addEdge(it->first,it->second);
        addEdge(it->second,it->first);
    }

    int ant=0;
    for(int i=1;i<=tot;i++)
    {
        ant=max(ant,solve(i));
    }
    cout<<ans<<endl<<ant<<endl;
}