计蒜客习题-最优订单方案 最大权闭合图

UPD:其实不需要getans函数 直接把所有正权点的点权加起来就是ans了

 

_______________________________________________________________

和NOI2006 最大获利相似的做法 但是这题要多dfs一遍求出选点的方案

从S出发dfs 如果e[i].c>0 就沿着这条链往下走 经过的点就是所选的点

算答案的时候是e[i^1].c>0 在剩余网络里找

 

#include<queue>
#include<vector>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
//#include<sstream>
const int maxn=207;
//正无穷定义为任意一个大于sigma W_i的整数
const int inf=0x3f3f3f3f;
struct edge{
    int v,c,nxt;
}e[maxn*maxn*2];
int p[maxn],eid=0,d[maxn],w[maxn],n,m,S,T;
std::vector<int> dd;
std::vector<int> jq;
bool bfs(){
    memset(d,-1,sizeof(d));
    std::queue<int> q;
    q.push(S);
    d[S]=0;
    while(!q.empty()){
        int u=q.front();q.pop();
        for(int i=p[u];~i;i=e[i].nxt){
            int v=e[i].v;
            if(e[i].c>0&&d[v]==-1){
                d[v]=d[u]+1;
                q.push(v);
            }
        }
    }
    return (d[T]!=-1);
}
int dfs(int u,int flow){
    if(u==T) return flow;
    int res=0,tmp;
    for(int i=p[u];~i;i=e[i].nxt){
        int v=e[i].v;
        if(e[i].c>0&&d[v]==d[u]+1)
        {   
            tmp=dfs(v,std::min(e[i].c,flow));
            res+=tmp;
            flow-=tmp;
            e[i].c-=tmp;
            e[i^1].c+=tmp;
            if(flow==0) break;//饱和
        }
    }    
    if(res==0) d[u]=-1;
    //printf("%d dfs\n",res);
    return res;    
}
int dinic(){
    int res=0;
    while(bfs()){
        res+=dfs(S,inf);
    }
    return res;
}
inline void ins(int u,int v,int c){
    e[eid].v=v;e[eid].c=c;e[eid].nxt=p[u];p[u]=eid++;
}
inline void add(int u,int v,int c){
    ins(u,v,c);ins(v,u,0);
}
inline void init(){
    memset(p,-1,sizeof(p));
    eid=0;
}
inline void read(int &x){
    x=0;
    char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)){ x=x*10+c-'0';c=getchar();}
}
bool vis[maxn];
int ans=0;
void find(int u){
    //if(u==T) return;
    for(int i=p[u];~i;i=e[i].nxt){
        int v=e[i].v;
        if(!vis[v]&&e[i].c>0){
            //原图中的边容量均为inf 必定连通 而
            if(w[v]>0) {
                dd.push_back(v);
            }else jq.push_back(v);
            vis[v]=true;
            find(v);
        }
    }
}
void getans(int u){
    for(int i=p[u];~i;i=e[i].nxt){
        int v=e[i].v;
        if(!vis[v]&&e[i^1].c>0){
            if(w[v]>0) ans+=w[v];
            vis[v]=true;
            getans(v);
        }
    }
}
int main(){
    read(n);
    read(m);    
    //printf("%d=n %d=m\n",n,m);
    /*
        每个订单，每台机器分别作为一个顶点
        如果订单A需要机器B add(a,b,inf) 订单点权为正数 机器点权为负数
        添加源点S与每个正权点v的有向边 add(S,i,w[i])
        汇点 add(i,T,-w[i])
    */
    init();
    S=0,T=n+m+1;
    for(int i=1;i<=n;i++){
        int t=0;
        read(w[i]);
        add(S,i,w[i]);
        char c=getchar();
        while(c!='\n'&&c!='\r'){
            while(c<'0'||c>'9') c=getchar();
            while(c>='0'&&c<='9') {
                t=t*10+c-'0';
                c=getchar();
            }
            add(i,t+n,inf);
            t=0;
        }
    }
    for(int i=1;i<=m;i++){
        read(w[i+n]);
        w[i+n]=-w[i+n];
        add(i+n,T,-w[i+n]);      
    }
    /*
    最小割是简单割 去掉最小割后的网络是闭合图 且该闭合图将达到最大权
    按照定义 闭合图的权和为正权点的权绝对值减去负权点的权的绝对值和
    */
    int res=dinic();
    vis[S]=vis[T]=true;
    find(S);
    memset(vis,false,sizeof(vis));
    getans(S);
    for(int i=0;i<dd.size();i++) printf("%d ",dd[i]);
    //std::cout<<dd.size()<<' '<<jq.size();
    printf("\n");
    for(int i=0;i<jq.size();i++){
        printf("%d ",jq[i]-n);
    }
    printf("\n%d",ans-res);
    return 0;
}