【ACM】2019杭电多校第二场1008 HDU6598 Harmonious Army

太菜了。。。

第一场的签到题是一道网络流，就想补网络流的题了。

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=6598

题目大意：

有n个士兵，其中m对有特殊能力，每个士兵有两种选择，Warrior或者Mage,当一对有特殊能力的士兵都为Warrior时，他们的贡献是a，若该对士兵都为Mage时，他们的贡献是c，其他情况下是b，求最大的贡献

解析:

这题官方题解讲的太详细了。。。

代码如下：

注意aa,bb,cc,dd,ee的类型如果为int的话结果不正确

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAXN=2010;
const int MAXM=1200010;
const double INF=0x3f3f3f3f;
struct Edge
{
    int to,next;
    double cap,flow;
}edge[MAXM];
int tol;
int head[MAXN];
void init()
{
    tol=2;
    memset(head,-1,sizeof(head));
}
void addedge(int u,int v,double w,double rw=0)
{
    edge[tol].to=v;
    edge[tol].cap=w;
    edge[tol].flow=0;
    edge[tol].next=head[u];
    head[u]=tol++;
    edge[tol].to=u;
    edge[tol].cap=rw;
    edge[tol].flow=0;
    edge[tol].next=head[v];
    head[v]=tol++;
}
int Q[MAXN];
int dep[MAXN],cur[MAXN],sta[MAXN];
bool bfs(int s,int t,int n)
{
    int front=0,tail=0;
    memset(dep,-1,sizeof(dep[0])*(n+1));
    dep[s]=0;
    Q[tail++]=s;
    while(front<tail)
    {
        int u=Q[front++];
        for(int i=head[u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].to;
            if(edge[i].cap>edge[i].flow&&dep[v]==-1)
            {
                dep[v]=dep[u]+1;
                if(v==t) return true;
                Q[tail++]=v;
            }
        }
    }
    return false;
}
double dinic(int s,int t,int n)
{
    double maxflow=0;
    while(bfs(s,t,n))
    {
        for(int i=0;i<n;i++) cur[i]=head[i];
        int u=s,tail=0;
        while(cur[s]!=-1)
        {
            if(u==t)
            {
                double tp=INF;
                for(int i=tail-1;i>=0;i--)
                    tp=min(tp,edge[sta[i]].cap-edge[sta[i]].flow);
                maxflow+=tp;
                for(int i=tail-1;i>=0;i--)
                {
                    edge[sta[i]].flow+=tp;
                    edge[sta[i]^1].flow-=tp;
                    if(edge[sta[i]].cap-edge[sta[i]].flow==0)
                        tail=i;
                }
                u=edge[sta[tail]^1].to;
            }
            else if(cur[u]!=-1&&edge[cur[u]].cap>edge[cur[u]].flow&&dep[u]+1==dep[edge[cur[u]].to])
            {
                sta[tail++]=cur[u];
                u=edge[cur[u]].to;
            }
            else
            {
                while(u!=s&&cur[u]==-1)
                    u=edge[sta[--tail]^1].to;
                cur[u]=edge[cur[u]].next;
            }
        }
    }
    return maxflow;
}
int main()
{
    //freopen("in.txt","r",stdin);
    int n,m;
    int u,v,a,b,c;
    while(~scanf("%d%d",&n,&m))
    {
        init();
        long long sum=0;
        while(m--)
        {
            scanf("%d%d%d%d%d",&u,&v,&a,&b,&c);
            double aa,bb,cc,dd,ee;
            aa=bb=(a+b)*1.0/2.0,cc=dd=(b+c)*1.0/2.0,ee=(a+c)*1.0/2.0-b;
            //源点为0，汇点为n+1
            addedge(0,u,aa);
            addedge(0,v,bb);
            addedge(u,n+1,cc);
            addedge(v,n+1,dd);
            addedge(u,v,ee);
            addedge(v,u,ee);
            sum+=(a+b+c);
        }
        printf("%lld\n",sum-(long long)dinic(0,n+1,n+2));
    }
    return 0;
}

新思路get√ 像这种需要划分成两个部分的题目，可以考虑网络流，构造一个S-T割(源点和汇点在两个不同的连通块里面的割)