ZOJ 有源汇有上下界的网络流

链接： https://cn.vjudge.net/problem/ZOJ-3229

题意：一个屌丝给m个女神拍照，他要拍n 天，每天要最多拍D 张照片，并且每天给 编号为i的女神拍的照片也有限制最小L最大R，并且每个女神要求她最后的照片数量不能小于G张。 问你有没有合理的方案，有合理的方案则输出每天屌丝给他要拍照的女神拍了多少张照片。

思路： 我们建立一个源点S和汇点T 对于每一天当做左边点，从S到天点建一条下限为0 上限为D的边，对于每个女神当做右边点，那么就可以从左边点向右边点建一条下限为L 上限为R的 边，那么最后再从女神点向汇点建一条下限为G上限inf 的边，然后就是一个裸的有源汇有上下界的网络流。

代码：

#include<bits/stdc++.h>

using namespace std;
const int N =1500;
const int inf =0x3f3f3f3f;

struct Edge
{
    int from,to,cap,flow;
    Edge(int u,int v,int c,int f):from(f),to(v),cap(c),flow(f){}
};
int n,m;


struct Dinic
{
    int n,m,s,t;
    vector< Edge >edges;
    vector< int >G[N];
    bool vis[N];
    int d[N];
    int cur[N];

    void init()
    {
        for (int i=0; i<N; i++)
            G[i].clear();
        edges.clear();
    }


    void addedge(int from,int to,int cap)
    {
        edges.push_back(Edge(from,to,cap,0));
        edges.push_back(Edge(to,from,0,0));
        m=edges.size();
        G[from].push_back(m-2);
        G[to].push_back(m-1);
        //cout<<"**** id "<<m-2<<endl;
    }

    bool bfs()
    {
        memset(vis,0,sizeof(vis));
        queue<int >q;
        for (int i=0; i<N; i++) d[i] = inf;
        q.push(s); d[s]=0;vis[s]=1;
        while(!q.empty()){
            int x=q.front(); q.pop();
            for(int i=0;i<G[x].size();i++){
                Edge e=edges[G[x][i]];
                if(!vis[e.to]&&e.cap>e.flow){
                    vis[e.to]=1;
                    d[e.to]=d[x]+1;
                    q.push(e.to);
                }
            }
        }
        return vis[t];
    }

    int dfs(int x,int a)
    {
        if(x==t||a==0) return a;
        int flow=0,f;
        for(int &i=cur[x];i<G[x].size();i++){
            Edge &e=edges[G[x][i]];
            if(d[x]+1==d[e.to]&&(f=dfs(e.to,min(a,e.cap-e.flow)))>0){
                e.flow+=f;
                edges[G[x][i]^1].flow-=f;
                flow+=f;
                a-=f;
                if(a==0) break;
            }
        }
        return flow;
    }

    int maxflow(int s,int t)
    {
        this->s=s; this->t=t;
        int flow=0;
        while(bfs()){
            memset(cur,0,sizeof(cur));
            flow+=dfs(s,inf);
        }

        return flow;
    }

}a;



int in[1500];
int out[1500];
int cnt[405];
int be[405][1005];
int D[N];
int G[N];

void init()
{
    a.init();
    memset(in,0,sizeof(in));
    memset(out,0,sizeof(out));
}

int main()
{
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        init();
        int S=n+m+1;
        int T=n+m+2;
        int SS=n+m+3;
        int TT=n+m+4;
        for(int i=1;i<=m;i++){
            scanf("%d",&G[i]);
        }

        int l,r,id;
        for(int i=1;i<=n;i++){
            scanf("%d %d",&cnt[i],&D[i]);
            for(int j=1;j<=cnt[i];j++){
                scanf("%d %d %d",&id,&l,&r);
                id+=(n+1);
                a.addedge(i,id,r-l);
                in[id]+=l;
                out[i]+=l;
                be[i][j]=l;
            }
        }

        for(int i=1;i<=m;i++){
            a.addedge(n+i,T,inf-G[i]);
            in[T]+=G[i];
            out[n+i]+=G[i];
        }

        for(int i=1;i<=n;i++){
            a.addedge(S,i,D[i]);
        }
        a.addedge(T,S,inf);

        for(int i=1;i<=n+m+2;i++){
            int jud=in[i]-out[i];
            if(jud>=0){
                a.addedge(SS,i,jud);
            }
            else{
                a.addedge(i,TT,-jud);
            }
        }
        a.maxflow(SS,TT);

        int f=0;
        for(int i=0;i<a.G[SS].size();i++){
            int id=a.G[SS][i];
            if(a.edges[id].flow!=a.edges[id].cap){
                f=1; break;
            }
        }
        if(f){
            printf("-1\n");
        }
        else{
            int Ans=a.maxflow(S,T);
            printf("%d\n",Ans);
            id=0;
            for(int i=1;i<=n;i++){
                for(int j=1;j<=cnt[i];j++){
                    id++;
                    int idd=(id-1)*2;
                    //cout<<"**** idd "<<idd<<endl;
                    //cout<<"below "<<be[i][j]<<endl;
                    printf("%d\n",a.edges[idd].flow+be[i][j]);
                }
            }
        }
        printf("\n");

    }
    return 0;
}

/*
1
4 6
1 2 1 3
2 3 1 3
3 4 1 3
4 1 1 3
1 3 1 3
4 2 1 3
*/