首页 > 代码库 > POJ 3680: Intervals【最小费用最大流】
POJ 3680: Intervals【最小费用最大流】
题目大意:你有N个开区间,每个区间有个重量wi,你要选择一些区间,使得满足:每个点被不超过K个区间覆盖的前提下,重量最大
思路:感觉是很好想的费用流,把每个区间首尾相连,费用为该区间的重量的相反数(由于要最大,所以是求最大费用最大流),容量为1,至于不超过K的限制,只要从源点到第一个点的流量为K就行,剩下每个相邻的点相连,费用为0,流量只要大于的等于K就可以(我取的正无穷)
//poj3680
#include <stdio.h>
#include <iostream>
#include <string.h>
#include <algorithm>
#include <queue>
#define maxn 120090
#define esp 0.00001
#define inf 0x3f3f3f3f
using namespace std;
int head[maxn],point[maxn],flow[maxn],next[maxn];
int now=0,value[maxn],k,xx,yy,vv,x[maxn],y[maxn];
int v[maxn],h=0,inte[maxn],id[maxn],root[maxn],n;
int dist[maxn],pre[maxn],j;
void add(int x,int y,int f,int v)
{
next[++now]=head[x];
head[x]=now;
point[now]=y;
flow[now]=f;
value[now]=v;
root[now]=x;
next[++now]=head[y];
head[y]=now;
point[now]=x;
flow[now]=0;
value[now]=-v;
root[now]=y;
}
int spfa(int s,int t)
{
for(int i=1;i<=j;i++)dist[i]=200000;
dist[t]=200000;
dist[s]=0;
int visit[maxn]={0};
visit[s]=1;
queue<int>q;
q.push(s);
while(!q.empty())
{
int u=q.front();
q.pop();
visit[u]=0;
for(int i=head[u];i;i=next[i])
{
int k=point[i];
if(dist[u]+value[i]<dist[k] && flow[i])
{
dist[k]=dist[u]+value[i];
pre[k]=i;
if(!visit[k])
{
visit[k]=1;
q.push(k);
}
}
}
}
if(dist[t]==200000)return 0;else return 1;
}
int main()
{
int tt;
scanf("%d",&tt);
while(tt--)
{
int ans=0;
now=0;h=0;
memset(head,0,sizeof(head));
memset(pre,0,sizeof(pre));
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
{
scanf("%d%d%d",&xx,&yy,&vv);
x[i]=xx;y[i]=yy;v[i]=vv;
inte[++h]=xx;inte[++h]=yy;
}
sort(inte+1,inte+1+h);
j=1;
id[inte[1]]=1;
for(int i=2;i<=h;i++)
{
if(inte[i]!=inte[j])
{
inte[++j]=inte[i];
id[inte[j]]=j;
}
}
for(int i=1;i<=j;i++)
add(i,i+1,inf,0);
for(int i=1;i<=n;i++)
{
add(id[x[i]],id[y[i]],1,-v[i]);
}
int s=maxn-10,t=maxn-100;
add(s,1,k,0);
add(j,t,k,0);
while(spfa(s,t))
{
int e=pre[t],minx=flow[e];
while(e)
{
minx=min(minx,flow[e]);
e=pre[root[e]];
}
e=pre[t];
while(e)
{
flow[e]-=minx;
flow[((e-1)^1)+1]+=minx;
e=pre[root[e]];
}
ans+=dist[t]*minx;
}
printf("%d\n",-ans);
}
return 0;
}
POJ 3680: Intervals【最小费用最大流】