HDU 4406 GPA（最大费用最大流）-CSDN博客

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4406

题意：有m科课程需要学习，每个课程有一个基础分数，每学习该课程一个时间单位，该课程的分数就增加1分。现在有n天的学习时间，每天有K个单位时间，并且每天可以学习的课程是固定的，给出学分绩点的计算方式，求可以达到的最高的学分绩点，要求所有课程都要及格。

思路：设f(x,w)=(4.0-3.0*(100-x)*(100-x)/1600)*w;

（1）每一天向汇点连边，流量K,费用0；

（2）每门课根据要求与天连边，流量K,费用0；

（3）源点向每门课连边：若基础分s小于60，则与源点连一条流量60-s、费用INF的边，保证优先到达60，接着从源点与该课程连

40条流量分别为1，费用为f(x+1,p)-f(x,p)的边（40条代表61到100）;若基础分s大于等于60，则连100-s条流量1、费用f(x+1,p)-f(x,p)的边。

求最大费用最大流。

完了统计从源点向各门课程的流量加上基础分，若还有小于60的，输出0；否则计算GPA。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#define min(x,y) ((x)<(y)?(x):(y))
using namespace std;


struct Node
{
    int u,v,flow,next;
    double cost;
};


const int INF=1000000000;
Node edges[40005];
int e,S,T;
int head[1005],in[1005],pre[1005],flow[1005];
double cost[1005];


void add(int u,int v,int flow,double cost)
{
    edges[e].u=u;
    edges[e].v=v;
    edges[e].flow=flow;
    edges[e].cost=cost;
    edges[e].next=head[u];
    head[u]=e++;
}


void Add(int u,int v,int flow,double cost)
{
    add(u,v,flow,cost);
    add(v,u,0,-cost);
}


int OK(int s,int t)
{
    int i,u,v,f;
    double c;
    queue<int> Q;

    memset(pre,-1,sizeof(pre));
    memset(in,0,sizeof(in));
    memset(flow,0,sizeof(flow));
    for(i=0;i<=t;i++) cost[i]=-1.0*INF;
    cost[s]=0;flow[s]=INF;Q.push(s);in[s]=1;
    while(!Q.empty())
    {
        u=Q.front();
        Q.pop();

        in[u]=0;
        for(i=head[u];i!=-1;i=edges[i].next)
        {
            v=edges[i].v;
            f=edges[i].flow;
            c=edges[i].cost;
            if(f>0&&cost[v]<cost[u]+c)
            {
                cost[v]=cost[u]+c;
                flow[v]=min(flow[u],f);
                pre[v]=i;
                if(!in[v]) Q.push(v),in[v]=1;
            }
        }
    }
    return flow[t];
}


int MCMF(int s,int t)
{
    int minflow,mincost=0,i;

    while(minflow=OK(s,t))
    {
        for(i=pre[t];i!=-1;i=pre[edges[i].u])
        {
            mincost+=minflow*edges[i].cost;
            edges[i].flow-=minflow;
            edges[i^1].flow+=minflow;
        }
    }
    return mincost;
}

int a[45][25],n,m,K,s[25],p[25];

double cal(int x,int w)
{
    return (4.0-3.0*(100-x)*(100-x)/1600)*w;
}

int main()
{
    while(scanf("%d%d%d",&n,&K,&m),n||m||K)
    {
        int i,j;
        for(i=1;i<=m;i++) scanf("%d",&p[i]);
        for(i=1;i<=m;i++) scanf("%d",&s[i]);
        for(i=1;i<=n;i++) for(j=1;j<=m;j++) scanf("%d",&a[i][j]);
        S=0;
        T=n+m+1;
        memset(head,-1,sizeof(head));
        e=0;
        for(i=1;i<=n;i++) Add(i+m,T,K,0);
        for(i=1;i<=n;i++) for(j=1;j<=m;j++) if(a[i][j])
            Add(j,i+m,K,0);
        double pre,cur;
        for(i=1;i<=m;i++)
        {
            if(s[i]<60)
            {
                Add(S,i,60-s[i],1.0*INF);
                pre=cal(60,p[i]);
                for(j=61;j<=100;j++)
                {
                    cur=cal(j,p[i]);
                    Add(S,i,1,cur-pre);
                    pre=cur;
                }
            }
            else
            {
                pre=cal(s[i],p[i]);
                for(j=s[i]+1;j<=100;j++)
                {
                    cur=cal(j,p[i]);
                    Add(S,i,1,cur-pre);
                    pre=cur;
                }
            }
        }
        MCMF(S,T);
        for(i=head[S];i!=-1;i=edges[i].next)
           s[edges[i].v]+=edges[i^1].flow;
        int sum=0;
        for(i=1;i<=m;i++) sum+=p[i];
        double ans=0;
        for(i=1;i<=m;i++)
        {
            if(s[i]<60) break;
            ans+=cal(s[i],p[i])/sum;
        }
        if(i<=m) ans=0;
        printf("%.6lf\n",ans);
    }
    return 0;
}