hdu 4494 Teamwork 网络流

网上比较多的代码都是用费用流做的,但是这个题目简单的描述就是求最小路径覆盖,每个点覆盖的次数不是1了而已。简单修改求最小路径覆盖的方法即可,源点向顶点,顶点向汇点连的边都变成点需要被覆盖的次数即可。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=15e1+9,inf=1e9;
int n,m;
int sum[maxn][10];
long long dist[maxn][maxn],x[maxn],y[maxn],st[maxn],ed[maxn];
int level[maxn<<1],que[maxn<<1];
int head[maxn<<1],lon;
struct
{
    int next,to,c;
}e[maxn*maxn<<2];
void edgeini()
{
    memset(head,-1,sizeof(head));
    lon=-1;
}
void edgemake(int from,int to,int c)
{
    e[++lon].c=c;
    e[lon].to=to;
    e[lon].next=head[from];
    head[from]=lon;
}
void make(int from,int to,int c)
{
    edgemake(from,to,c);
    edgemake(to,from,0);
}

bool makelevel(int s,int t)
{
    memset(level,0,sizeof(level));
    int front=1,end=0;
    que[++end]=s;
    level[s]=1;
    while(front<=end)
    {
        int u=que[front++];
        if(u==t) return true;
        for(int k=head[u];k!=-1;k=e[k].next)
        {
            int v=e[k].to;
            if(!level[v]&&e[k].c)
            {
                que[++end]=v;
                level[v]=level[u]+1;
            }
        }
    }
    return false;
}

int dfs(int now,int t,int maxf)
{
    if(now==t) return maxf;
    int ret=0;
    for(int k=head[now];k!=-1;k=e[k].next)
    {
        int u=e[k].to;
        if(level[u]==level[now]+1&&e[k].c)
        {
            int f=dfs(u,t,min(e[k].c,maxf-ret));
            e[k].c-=f;
            e[k^1].c+=f;
            ret+=f;
            if(ret==maxf) return ret;
        }
    }
    return ret;
}

int maxflow(int s,int t)
{
    int ret=0;
    while(makelevel(s,t))
    {
        ret+=dfs(s,t,inf);
    }
    return ret;
}


void getdist()
{
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    {
        dist[i][j]=(x[i]-x[j])*(x[i]-x[j]);
        dist[i][j]+=(y[i]-y[j])*(y[i]-y[j]);
    }
}

void makegraph(int tt)
{
    edgeini();
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    if(i!=j)
    {
        if(st[j]>=ed[i])
        {
            long long tmp=(st[j]-ed[i])*(st[j]-ed[i]);
            if(tmp>=dist[i][j])
            {
                make(i,j+n,inf);
            }
        }
    }
    for(int i=1;i<=n;i++)
    {
        make(0,i,sum[i][tt]);
        make(i+n,n+n+1,sum[i][tt]);
    }
}

int main()
{
//    freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int ans=0;
        scanf("%d%d",&n,&m);
        scanf("%d%d",&x[1],&y[1]);
        n--;
        for(int i=1;i<=n;i++)
        {
            scanf("%I64d %I64d %I64d %I64d",&x[i],&y[i],&st[i],&ed[i]);
            ed[i]+=st[i];
            for(int j=1;j<=m;j++)
            {
                scanf("%d",&sum[i][j]);
                ans+=sum[i][j];
            }
        }
        getdist();
        for(int i=1;i<=m;i++)
        {
            makegraph(i);
            ans-=maxflow(0,n+n+1);
        }
        cout<<ans<<endl;
    }
    return 0;
}


你可能感兴趣的:(hdu 4494 Teamwork 网络流)