HDU 4975 (杭电多校 #10 1005题）A simple Gaussian elimination problem.（网络流之最大流）

题目地址：HDU 4975

对这题简直无语。。。本来以为这题要用什么更先进的方法，结果还是老方法，这么卡时间真的好吗。。。。比赛的时候用了判环的方法，一直TLE。。后来换了矩阵DP的方式，加了加剪枝就过了。。无语了。。

代码如下：

[cpp]
view plaincopyprint?





#include <cstdio> 

#include <cstring> 
#include <algorithm> 

#include <iostream> 
#include <cstdio> 

#include <string> 
#include <cstring> 

#include <stdlib.h> 
#include <math.h> 

#include <ctype.h> 
#include <queue> 

#include <map> 
#include <set> 

#include <algorithm> 
 
using namespace std; 

const int INF=0x3f3f3f3f; 

int head[1010], source, sink, nv, cnt, vis[610][600], mp[600][600]; 

int cur[1010], num[1010], d[1010], pre[1010]; 

int a[600], b[600]; 
struct node 
{ 
    int u, v, cap, next; 

} edge[1000000]; 
void add(int u,
int v, int cap) 

{ 
    edge[cnt].v=v; 
    edge[cnt].cap=cap; 
    edge[cnt].next=head[u]; 
    head[u]=cnt++; 
 
    edge[cnt].v=u; 
    edge[cnt].cap=0; 
    edge[cnt].next=head[v]; 
    head[v]=cnt++; 
} 
void bfs() 
{ 
    memset(num,0,sizeof(num)); 

    memset(d,-1,sizeof(d)); 

    queue<int>q; 

    q.push(sink); 
    d[sink]=0; 
    num[0]=1; 
    while(!q.empty()) 

    { 
        int u=q.front(); 

        q.pop(); 
        for(int i=head[u]; i!=-1; i=edge[i].next) 

        { 
            int v=edge[i].v; 

            if(d[v]==-1) 
            { 
                d[v]=d[u]+1; 
                num[d[v]]++; 
                q.push(v); 
            } 
        } 
    } 
} 
int isap() 
{ 
    memcpy(cur,head,sizeof(cur)); 

    int flow=0, u=pre[source]=source, i; 

    bfs(); 
    while(d[source]<nv) 
    { 
        if(u==sink) 
        { 
            int f=INF, pos; 

            for(i=source; i!=sink; i=edge[cur[i]].v) 

            { 
                if(f>edge[cur[i]].cap) 

                { 
                    f=edge[cur[i]].cap; 
                    pos=i; 
                } 
            } 
            for(i=source; i!=sink; i=edge[cur[i]].v) 

            { 
                edge[cur[i]].cap-=f; 
                edge[cur[i]^1].cap+=f; 
            } 
            flow+=f; 
            u=pos; 
        } 
        for(i=cur[u]; i!=-1; i=edge[i].next) 

        { 
            if(d[edge[i].v]+1==d[u]&&edge[i].cap) 

                break; 
        } 
        if(i!=-1) 
        { 
            cur[u]=i; 
            pre[edge[i].v]=u; 
            u=edge[i].v; 
        } 
        else 
        { 
            if(--num[d[u]]==0)
break; 
            int mind=nv; 

            for(i=head[u]; i!=-1; i=edge[i].next) 

            { 
                if(mind>d[edge[i].v]&&edge[i].cap) 

                { 
                    mind=d[edge[i].v]; 
                    cur[u]=i; 
                } 
            } 
            d[u]=mind+1; 
            num[d[u]]++; 
            u=pre[u]; 
        } 
    } 
    return flow; 

} 
bool panduan(int n,int m) 

{ 
    memset(vis,0,sizeof(vis)); 

    int i, j, k; 
    for(i=1;i<=n;i++) 

    { 
        if(a[i]==0||a[i]==9*m)
continue ; 
        for(j=1;j<=m;j++) 
        { 
            if(b[j]==0||b[j]==9*n)
continue ; 
            for(k=j+1;k<=m;k++) 

            { 
                int t1=0, t2=0; 

                if(mp[i][j]&&mp[i][k]!=9) 

                { 
                    if(vis[j][k])
return 1; 
                    t1=1; 
                } 
                if(mp[i][k]&&mp[i][j]!=9) 

                { 
                    if(vis[k][j])
return 1; 
                    t2=1; 
                } 
                if(t1) vis[k][j]=1; 

                if(t2) vis[j][k]=1; 

            } 
        } 
    } 
    return 0; 
} 
int read() 
{ 
    int x = 0; 

    char ch = ' '; 

    while(ch < '0' || ch >
'9') ch = getchar(); 
    while(ch >= '0' && ch <=
'9') x = x * 10 + ch - '0', ch = getchar(); 

    return x; 
} 
int main() 
{ 
    int t, i, j, n, m, sum1, sum2, num=0; 

    t=read(); 
    while(t--) 
    { 
        num++; 
        n=read(); 
        m=read(); 
        sum1=sum2=0; 
        memset(head,-1,sizeof(head)); 

        cnt=0; 
        source=0; 
        sink=n+m+1; 
        nv=sink+1; 
        for(i=1; i<=n; i++) 

        { 
            a[i]=read(); 
            sum1+=a[i]; 
        } 
        for(i=1; i<=m; i++) 

        { 
            b[i]=read(); 
            sum2+=b[i]; 
        } 
        printf("Case #%d: ",num); 

        if(sum1!=sum2) 

        { 
            puts("So naive!"); 

            continue ; 
        } 
        for(i=1; i<=n; i++) 

        { 
            add(source,i,a[i]); 
            for(j=1; j<=m; j++) 

            { 
                add(i,j+n,9); 
            } 
        } 
        for(i=1; i<=m; i++) 

        { 
            add(i+n,sink,b[i]); 
        } 
        int ans=isap(); 
        if(ans!=sum1) 

        { 
            puts("So naive!"); 

            continue ; 
        } 
        for(i=1;i<=n;i++) 
        { 
            for(j=head[i];j!=-1;j=edge[j].next) 

            { 
                int v=edge[j].v; 

                if(v>n&&v<=n+m) 

                    mp[i][v-n]=9-edge[j].cap; 
            } 
        } 
        if(panduan(n,m)) 

        { 
            puts("So young!"); 

        } 
        else 
            puts("So simple!"); 

    } 
    return 0; 
}