【网络流24题】最长k可重线段集（费用流）

题面

Cogs的数据有问题

Loj

洛谷

题解

这道题和最长k可重区间集没有区别

只不过费用额外计算一下

但是，还是有一点要注意的地方

这里可以是一条垂直的直线

所以，首先把所有的x轴全部乘2

如果两个相等就把右端点+1

否则左端点+1

这样就可以解决垂直于x轴的问题了

#include<iostream>

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<cmath>

#include<algorithm>

#include<set>

#include<map>

#include<vector>

#include<queue>

using namespace std;

#define MAX 50000

#define MAXL 5000000

#define INF 1000000000

inline int read()

{

	int x=0,t=1;char ch=getchar();

	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();

	if(ch=='-')t=-1,ch=getchar();

	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();

	return x*t;

}

struct Line

{

    int v,next,w,fy;

}e[MAXL];

bool vis[MAX];

int h[MAX],cnt=2;

inline void Add(int u,int v,int w,int fy)

{

    e[cnt]=(Line){v,h[u],w,fy};h[u]=cnt++;

    e[cnt]=(Line){u,h[v],0,-fy};h[v]=cnt++;

}

int pe[MAX],pr[MAX],dis[MAX];

int S,T,Cost,n,m,Flow,opt=-1;

bool SPFA()

{

    memset(dis,63,sizeof(dis));

    queue<int> Q;

    Q.push(S);dis[S]=0;

    while(!Q.empty())

    {

        int u=Q.front();Q.pop();

        for(int i=h[u];i;i=e[i].next)

        {

            int v=e[i].v;

            if(e[i].w&&dis[v]>dis[u]+e[i].fy)

            {

                dis[v]=dis[u]+e[i].fy;

                pe[v]=i;pr[v]=u;

                if(!vis[v])vis[v]=true,Q.push(v);

            }

        }

        vis[u]=false;

    }

    if(dis[T]>=INF)return false;

    int flow=INF;

    for(int i=T;i!=S;i=pr[i])flow=min(flow,e[pe[i]].w);

    for(int i=T;i!=S;i=pr[i])e[pe[i]].w-=flow,e[pe[i]^1].w+=flow;

    Cost+=opt*flow*dis[T];

    Flow+=flow;

    return true;

}

int X1[MAX],X2[MAX],SS[MAX],tot,K,W[MAX];

int main()

{

	n=read();K=read();

	for(int i=1;i<=n;++i)

	{

		int x1=read(),y1=read(),x2=read(),y2=read();

		int ww=sqrt(1ll*(x1-x2)*(x1-x2)+1ll*(y1-y2)*(y1-y2));

		if(x1>x2)swap(x1,x2);

		x1*=2;x2*=2;if(x1==x2)x2++;else x1++;

		SS[++tot]=x1;SS[++tot]=x2;

		X1[i]=x1,X2[i]=x2;W[i]=ww;

	}

	sort(&SS[1],&SS[tot+1]);

	tot=unique(&SS[1],&SS[tot+1])-SS-1;

	S=0;T=tot+1;

	for(int i=0;i<=tot;++i)

		Add(i,i+1,K,0);

	for(int i=1;i<=n;++i)

	{

		int u=lower_bound(&SS[1],&SS[tot+1],X1[i])-SS;

		int v=lower_bound(&SS[1],&SS[tot+1],X2[i])-SS;

		Add(u,v,1,-W[i]);

	}

	while(SPFA());

	printf("%d\n",Cost);

	return 0;

}