BZOJ1497 [NOI2006] 最大获利

1497: [NOI2006]最大获利

Time Limit: 5 Sec Memory Limit: 64 MB
Submit: 5147 Solved: 2506
Description
新的技术正冲击着手机通讯市场，对于各大运营商来说，这既是机遇，更是挑战。THU集团旗下的CS&T通讯公司在新一代通讯技术血战的前夜，需要做太多的准备工作，仅就站址选择一项，就需要完成前期市场研究、站址勘测、最优化等项目。在前期市场调查和站址勘测之后，公司得到了一共N个可以作为通讯信号中转站的地址，而由于这些地址的地理位置差异，在不同的地方建造通讯中转站需要投入的成本也是不一样的，所幸在前期调查之后这些都是已知数据：建立第i个通讯中转站需要的成本为Pi（1≤i≤N）。另外公司调查得出了所有期望中的用户群，一共M个。关于第i个用户群的信息概括为Ai, Bi和Ci：这些用户会使用中转站Ai和中转站Bi进行通讯，公司可以获益Ci。（1≤i≤M, 1≤Ai, Bi≤N） THU集团的CS&T公司可以有选择的建立一些中转站（投入成本），为一些用户提供服务并获得收益（获益之和）。那么如何选择最终建立的中转站才能让公司的净获利最大呢？（净获利 = 获益之和 - 投入成本之和）
Input
输入文件中第一行有两个正整数N和M 。第二行中有N个整数描述每一个通讯中转站的建立成本，依次为P1, P2, …, PN 。以下M行，第(i + 2)行的三个数Ai, Bi和Ci描述第i个用户群的信息。所有变量的含义可以参见题目描述。
Output
你的程序只要向输出文件输出一个整数，表示公司可以得到的最大净获利。
Sample Input
5 5
1 2 3 4 5
1 2 3
2 3 4
1 3 3
1 4 2
4 5 3
Sample Output
4
HINT
【样例说明】选择建立1、2、3号中转站，则需要投入成本6，获利为10，因此得到最大收益4。【评分方法】本题没有部分分，你的程序的输出只有和我们的答案完全一致才能获得满分，否则不得分。【数据规模和约定】 80%的数据中：N≤200，M≤1 000。 100%的数据中：N≤5 000，M≤50 000，0≤Ci≤100，0≤Pi≤100。

分析：
将中转站、用户群看成点，Ai表示中转站，Bi表示用户群
从源点向中转站连边权值是建立中转站的费用，割掉这条边表示建立该中转站。建立点Bi，连边向T，权值为满足该用户之后可以获得的收益，割掉这条边表示不满足该用户的需求，损失收益Ci。如果第i个用户会使用第j个中转站，那么从S向Aj的边和Bi向T的边就不能同时保留（因为我们已经定义了割掉从S向Aj的边才表示建立中转站j，保留Bi向T的边表示满足了第i个用户的需求[挣到钱了]，那么这显然是冲突的，没有建立中转站却又挣到了钱~~），那么怎么解决呢。。。
解决方案：从Aj向Bi连一条权值为INF的边，这样就可以保证两条边不会同时存在了。
所有用户的收益之和减去最小割就是最大净获利了。

/**************************************************************
    Problem: 1497
    User: Twi_etn
    Language: C++
    Result: Accepted
    Time:280 ms
    Memory:13920 kb
****************************************************************/

#include<iostream>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
#define maxm 1000000
#define INF 0x3fffffff
#define maxn 100000
int n,m,T,S,dep[maxn],head[maxn],tot=1;
struct Edge{ int to,value,next; }e[maxm];
int read(){
    int w=0,c=1;char ch=getchar();
    while(ch<'0'||ch>'9'){ if(ch=='-') c=-1; ch=getchar(); }
    while(ch>='0' && ch<='9') w=w*10+ch-'0',ch=getchar();
    return w*c;
}
void Add_Edge(int u,int v,int w){
    tot++;e[tot].to=v;e[tot].value=w;
    e[tot].next=head[u];head[u]=tot;
}
queue<int> q;
bool BFS(){
    memset(dep,-1,sizeof dep );
    while(!q.empty()) q.pop();
    dep[S]=0;q.push(S);
    while(!q.empty()){
        int u=q.front();q.pop();
        for(int i=head[u];i;i=e[i].next){
            int v=e[i].to;
            if(dep[v]==-1&&e[i].value){
                dep[v]=dep[u]+1;
                q.push(v);
                if(v==T) return true;
            }
        }
    }
    return false;
}
int DFS(int u,int flow){
    int ret=0;
    if(u==T) return flow;
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to;
        if(dep[v]==dep[u]+1&&e[i].value){
            int x=DFS(v,min(flow-ret,e[i].value));
            ret+=x;e[i].value-=x;e[i^1].value+=x;
            if(ret==flow) return flow;
        }
    }
    if(!ret) dep[u]=-1;
    return ret;
}
int Dinic(){
    int ans=0;
    while(BFS()) {
        int alpq=0;
        while(alpq=DFS(S,INF)) ans+=alpq;
    }
    return ans;
}
int main(){
    n=read();m=read();
    T=n+m+1;S=0;
    for(int x,i=1;i<=n;i++){
        x=read();
        Add_Edge(S,i,x);Add_Edge(i,S,0);
    }
    int sum=0;
    for(int i=1,u,v,w;i<=m;i++){
        u=read();v=read();w=read();sum+=w;
        Add_Edge(u,i+n,INF);Add_Edge(i+n,u,0);
        Add_Edge(v,i+n,INF);Add_Edge(i+n,v,0);
        Add_Edge(i+n,T,w);  Add_Edge(T,i+n,0);
    }
    printf("%d\n",sum-Dinic());
    return 0;
}

时隔多年，再次回来做这道题

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
#define MAXN 258005
#define MAXM 500005//试了好几遍 才AC 刚开始RE两组 
#define INF 0x3fffffff
#define LL long long
struct Edge{ int to,value,next; }e[MAXM];
int dep[MAXN],head[MAXN],T,S,n,m,tot=1;
bool exist[MAXN];
LL sum=0;
inline int read(int &x){
    x=0;int f=1;char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+(c-'0');c=getchar();}
    x*=f;
}
inline void Add_Edge(int u,int v,int w){
    e[++tot].to=v;e[tot].value=w;
    e[tot].next=head[u];head[u]=tot;
}
queue<int> q;
bool BFS(){
    memset(dep,-1,sizeof dep );
    memset(exist,false,sizeof exist );
    dep[S]=1;q.push(S);
    while(!q.empty()){
        int u=q.front();q.pop();exist[u]=false;
        for(int i=head[u];i;i=e[i].next){
            int v=e[i].to;
            if(e[i].value&&dep[v]==-1){
                dep[v]=dep[u]+1;
                if(!exist[v]){ q.push(v);exist[v]=true; }
            }
        }
    }
    if(dep[T]!=-1) return true;
    else return false;
}
int DFS(int u,int flow){
    if(u==T) return flow;
    int ret=0;
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to;
        if(dep[v]==dep[u]+1&&e[i].value){
            int x=DFS(v,min(flow-ret,e[i].value));
            ret+=x;e[i].value-=x;e[i^1].value+=x;
            if(ret==flow) return flow;
        }
    }
    return ret;
}
int Dinic(){
    int ans=0;
    while(BFS()) ans+=DFS(S,INF);
    return ans;
}
int main(){
    read(n);read(m);
    T=n+m+1;S=0;
    for(int x,i=1;i<=n;i++){//每个中转站
        read(x);
        Add_Edge(i,T,x);Add_Edge(T,i,0);
    }
    for(int u,v,w,i=1;i<=m;i++){//每个用户
        read(u);read(v);read(w);sum+=w;
        Add_Edge(S,i+n,w);Add_Edge(i+n,S,0);
        Add_Edge(i+n,u,INF);Add_Edge(u,i+n,0);
        Add_Edge(i+n,v,INF);Add_Edge(v,i+n,0);
    }
    printf("%lld\n",sum-Dinic());
    return 0;
}

用户和S点连边，中转站和T点连边，看我原来的代码能看出来我原来的代码中转站和S点连了边，然而我再次评测一遍，新代码跑0.5s，旧的跑0.1s，正确性是都没有问题的。。。不解。

最大权闭合子图问题->最小割->最小割最大流定理->最大流->Dinic