网络流24题之最长k可重线段集问题

对于每个线段拆成两个点，如同之前一样建图，由于可能出现垂直于x轴的

所以建图由i指向i~

继续最小费用最大流

By：大奕哥

 #include<bits/stdc++.h>
 using namespace std;
 const int N=,inf=1e9;
 int head[N],d[N],f[N],l1[N],r1[N],l2[N],r2[N],a[N],s=1e9,t,n,k,cnt=-;
 long long cost;
 bool v[N];
 struct node{
     int to,nex,f,w,c;
 }e[];
 void add(int x,int y,int w,int c)
 {
     e[++cnt].to=y;e[cnt].w=w;e[cnt].f=x;e[cnt].c=c;e[cnt].nex=head[x];head[x]=cnt;
     e[++cnt].to=x;e[cnt].w=;e[cnt].f=y;e[cnt].c=-c;e[cnt].nex=head[y];head[y]=cnt;
 }
 queue<int>q;
 bool spfa()
 {
     memset(f,-,sizeof(f));
     memset(d,0x3f,sizeof(d));
     memset(v,,sizeof(v));
     d[s]=;v[s]=;q.push(s);
     while(!q.empty())
     {
         int x=q.front();q.pop();v[x]=;
         for(int i=head[x];i!=-;i=e[i].nex)
         {
             int y=e[i].to;
             if(d[y]<=d[x]+e[i].c||!e[i].w)continue;

             d[y]=d[x]+e[i].c;f[y]=i;
             if(!v[y])q.push(y),v[y]=;
         }
     }
     if(d[t]>1e9)return ;
     int flow=inf;
     for(int i=f[t];i!=-;i=f[e[i].f])
     flow=min(flow,e[i].w);
     for(int i=f[t];i!=-;i=f[e[i].f])
     e[i].w-=flow,e[i^].w+=flow,cost+=1ll*e[i].c*flow;
     return ;
 }
 int main()
 {
     scanf("%d%d",&n,&k);int num=;
     memset(head,-,sizeof(head));
     for(int i=;i<=n;++i)
     {
         scanf("%d%d%d%d",&l1[i],&r1[i],&l2[i],&r2[i]);
         a[++num]=l1[i];a[++num]=l2[i];
     }
     sort(a+,a++num);
     num=unique(a+,a++num)-a-;
     for(int i=;i<=n;++i)
     {
         int x=sqrt(1ll*(l1[i]-l2[i])*(l1[i]-l2[i])+1ll*(r2[i]-r1[i])*(r2[i]-r1[i]));
         l1[i]=lower_bound(a+,a++num,l1[i])-a;
         l2[i]=lower_bound(a+,a++num,l2[i])-a;
         if(l1[i]!=l2[i])
         add((l1[i]<<)|,l2[i]<<,,-x);
         else
         add(l1[i]<<,(l2[i]<<)|,,-x);
     }
     for(int i=;i<num;++i)
     {
         add((i<<)|,i+<<,inf,);
         add(i<<,(i<<)|,inf,);
     }
     add(num<<,(num<<)|,inf,);
     t=num*+;
     add((num<<)|,t,k,);
     add(,,k,);s=;
     while(spfa());
     printf("%lld\n",-cost);
     return ;
 }