hdu4975 A simple Gaussian elimination problem.(多校第十场1005）

此题与hdu4888类似，但由于数据范围比较大，所以用4888的代码会T，关键在于判环，可以用vis[i][j][0]表示在某一行是i,j列可以变小，vis[i][j][1]表示可以变大，然后，在一行上扫时可以用vector记录0，9，非0、9的位置。

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<string.h>
#include<queue>
#define inf 100000000
#define maxn 1050
using namespace std;
struct pi
{
    int to;
    int cost;
    int rev;
}pp;
vector<pi >g[maxn];
vector<int >gg[3];
queue<int>q;
int a[505][505];
bool vis[505][505][2];
int s,t;
int line[maxn],leve[maxn];
int min(int a,int b)
{
    int p;
    p=a;
    if(b<a)
        p=b;
    return p;
}
void add(int a,int b,int cos)
{
    pp.to=b;
    pp.cost=cos;
    pp.rev=(int)g[b].size();
    g[a].push_back(pp);
    pp.to=a;
    pp.cost=0;
    pp.rev=(int)g[a].size()-1;
    g[b].push_back(pp);
    return ;
}
void bfs(void)
{
    q.push(s);
    int p,f,i;
    while(!q.empty())
    {
        p=q.front();
        q.pop();
        f=(int)g[p].size();
        for(i=0;i<f;i++)
        {
            pi &e=g[p][i];
            if(e.cost>0&&line[e.to]<0)
            {
                line[e.to]=line[p]+1;
                q.push(e.to);
            }
        }
    }
    return ;
}
int dfs(int v,int f)
{
    int p;
    if(v==t)
        return f;
    for(int &i=leve[v];i<g[v].size();i++)
    {
        pi &e=g[v][i];
        if(line[v]<line[e.to]&&e.cost>0)
        {
            p=dfs(e.to,min(f,e.cost));
            if(p>0)
            {
                e.cost-=p;
                g[e.to][e.rev].cost+=p;
                return p;
            }
        }
    }
    return 0;
}
int  dinic()
{
    int f;
    int flow=0;
    while(1)
    {
        memset(line,-1,sizeof(line));
        memset(leve,0,sizeof(leve));
        line[s]=0;
        bfs();
        if(line[t]<0)
            return flow;
        f=dfs(s,inf);
        if(f==0)
            continue;
        flow+=f;
        while((f=dfs(s,inf))>0)
        {
            flow+=f;
        }
    }
}
int main()
{
    int i,j,n,m,p,tt,N,s1,s2,flag,k;
    cin>>tt;
    N=tt;
    while(tt--){
        scanf("%d%d",&n,&m);
        for(i=0;i<=n+m+1;i++) g[i].clear();
        s1=0;
        s2=0;
        for(i=1;i<=n;i++){
            scanf("%d",&p);
            add(0,i,p);
            s1+=p;
        }
        for(i=1;i<=m;i++){
            scanf("%d",&p);
            add(n+i,n+m+1,p);
            s2+=p;
        }
        if(s1!=s2){
            printf("Case #%d: So naive!\n",N-tt);
            continue;
        }
        for(i=1;i<=n;i++){
            for(j=1;j<=m;j++){
                add(i,n+j,9);
            }
        }
        s=0;
        t=n+m+1;
        if(dinic()!=s1){
            printf("Case #%d: So naive!\n",N-tt);
            continue;
        }
        for(i=1;i<=n;i++){
            p=g[i].size();
            for(j=0;j<p;j++){
                if(g[i][j].to<=n) continue;
                a[i][g[i][j].to-n]=9-g[i][j].cost;
            }
        }
        flag=0;
        memset(vis,0,sizeof(vis));
        for(i=1;i<=n;i++){
            for(j=0;j<3;j++) gg[j].clear();
            for(j=1;j<=m;j++){
                if(a[i][j]==9){
                    p=gg[0].size();
                    for(k=0;k<p;k++){
                        if(vis[gg[0][k]][j][0]){
                            flag=1;
                            break;
                        }
                        vis[gg[0][k]][j][1]=1;
                    }
                    if(flag) break;
                    p=gg[1].size();
                    for(k=0;k<p;k++){
                        if(vis[gg[1][k]][j][0]){
                            flag=1;
                            break;
                        }
                        vis[gg[1][k]][j][1]=1;
                    }
                    gg[2].push_back(j);
                }
                else if(a[i][j]==0){
                    p=gg[1].size();
                    for(k=0;k<p;k++){
                        if(vis[gg[1][k]][j][1]){
                            flag=1;
                            break;
                        }
                        vis[gg[1][k]][j][0]=1;
                    }
                    if(flag) break;
                    p=gg[2].size();
                    for(k=0;k<p;k++){
                        if(vis[gg[2][k]][j][1]){
                            flag=1;
                            break;
                        }
                        vis[gg[2][k]][j][0]=1;
                    }
                    gg[0].push_back(j);
                }
                else{
                    p=gg[0].size();
                    for(k=0;k<p;k++){
                        if(vis[gg[0][k]][j][0]){
                            flag=1;
                            break;
                        }
                        vis[gg[0][k]][j][1]=1;
                    }
                    if(flag) break;
                    p=gg[2].size();
                    for(k=0;k<p;k++){
                        if(vis[gg[2][k]][j][1]){
                            flag=1;
                            break;
                        }
                        vis[gg[2][k]][j][0]=1;
                    }
                    if(flag) break;
                    p=gg[1].size();
                    for(k=0;k<p;k++){
                        if(vis[gg[1][k]][j][1]||vis[gg[1][k]][j][0]){
                            flag=1;
                            break;
                        }
                        vis[gg[1][k]][j][0]=1;
                        vis[gg[1][k]][j][1]=1;
                    }
                    gg[1].push_back(j);
                }
                if(flag) break;
            }
            if(flag) break;
        }
        if(flag){
            printf("Case #%d: So young!\n",N-tt);
        }
        else{
            printf("Case #%d: So simple!\n",N-tt);
        }
    }
}