洛谷 P3358 最长k可重区间集问题

题面

题意

在一条线上有n个开区间,现在要选择一些,使每个点最多被k个区间覆盖且开区间总长度(右端点减左端点)最大.

方法

这题难与限制每个点的区间覆盖数,无法将最长长度转化为最大流,但可以通过最大费用流(将所有边的费用取反再跑一遍最小费用流)来做.
具体做法是相邻点连一条费用为0,流量为INF的边,区间两端点连一条费用为区间长度,流量为1的边,s到1与最右端点到t连一条费用为0,流量为k的边.
这样可以用最大流保证每个点最多被k个区间覆盖,用费用流保证总长度最大.
另外要注意的是不属于区间端点的点不需要建出来,点数过多会增大常数.

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
#define INF 0x3f3f3f3f
#define N 100000
using namespace std;

int n,m,bb=1,first[N+100],s,t,ans,d[N+100],B[N+100],last[N+100],num[N];
bool in[N+100];
struct Bn
{
    int to,next,quan,cst;
}bn[(N+100)*2];
queue<int>que;

inline void add(int u,int v,int w,int z)
{
    bb++;
    bn[bb].to=v;
    bn[bb].next=first[u];
    bn[bb].cst=z;
    bn[bb].quan=w;
    first[u]=bb;
}

inline void ad(int u,int v,int w,int z)
{
    add(u,v,w,z);
    add(v,u,0,-z);
}

int dfs(int now,int mn)
{
    if(now==s)
        return mn;
    int res;
    res=dfs(last[now],min(bn[B[now]].quan,mn));
    bn[B[now]].quan-=res;
    bn[B[now]^1].quan+=res;
    return res;
}

inline bool bfs()
{
    int p,q;
    memset(d,INF,sizeof(d));
    que.push(s);
    d[s]=0;
    for(;!que.empty();)
    {
        q=que.front();
        que.pop();
        in[q]=0;
        for(p=first[q];p!=-1;p=bn[p].next)
        {
            if(d[bn[p].to]<=d[q]+bn[p].cst||!bn[p].quan) continue;
            d[bn[p].to]=d[q]+bn[p].cst;
            last[bn[p].to]=q;
            B[bn[p].to]=p;
            if(!in[bn[p].to])
            {
                in[bn[p].to]=1;
                que.push(bn[p].to);
            }
        }
    }
    if(d[t]==INF) return 0;
    ans-=d[t]*dfs(t,INF);
    return 1;
}

int main()
{
    memset(first,-1,sizeof(first));
    int i,j,p,q;
    cin>>n>>m;
    t=N+1;
    for(i=1;i<=n;i++)
    {
        scanf("%d%d",&p,&q);
        num[i*2-1]=p;
        num[i*2]=q;
        ad(p,q,1,p-q);
    }
    sort(num+1,num+n*2+1);
    for(i=1;i<2*n;i++)
    {
        ad(num[i],num[i+1],INF,0);
    }
    ad(s,num[1],m,0);
    ad(num[n*2],t,m,0);
    for(;bfs(););
    cout<<ans;
}