关闭

POJ 3680

标签: struct算法
1859人阅读 评论(0) 收藏 举报
分类:

建图很巧妙

先离散化区间端点。从0->1,1->2,~~~~~n->n+1每条边容量为k,费用为0,

对于每条线段,他的两个端点连边容量为1,费用-w

然后跑一遍最小费用流

算法正确性证明:

如果两个区间没有交集,那么代表它们的边可以出现在同一增广路上,这一点显然。否则,它们就在不同的增广路上。每一个区间对应的边的容量都是1,这样,最后的流量就是被选出的两两有交集的区间的数量。受到(0,1,k,0)这条边的容量限制,这个值刚好不大于k.区间的权都是正的,因此选取的区间多多益善,所以流量必然最大。

 (对于每次选取的增广路中总存在一个区间,在每次增广所得区间都与这个区间有交集)

#include<cstdio>
#include<stdlib.h>
#include<cstring>
using namespace std;
const int inf=99999999;
struct{
    int v, cap, cost, next, re;
}edge[2005];
struct{
	int a,b,w;
}line[210];
struct Li{
	int be,num;
}li[450];
int n,ans;
int k,edgeHead[500];
int que[500],pre[500],dis[500];
bool vis[500];
int cmp(const void * a,const void *b){  
    struct Li *aa=(struct Li *) a;  
    struct Li *bb=(struct Li *) b;  
	return aa->be-bb->be;  
}  

void addEdge(int u,int v,int ca,int co){
    edge[k].v=v;
    edge[k].cap=ca;
    edge[k].cost=co;
    edge[k].next=edgeHead[u];
    edge[k].re=k + 1;               //这个用来记录此边的反边
    edgeHead[u]=k ++;
    edge[k].v=u;
    edge[k].cap=0;
    edge[k].cost=-co;               //故这里去反费用(因为是反向边)
    edge[k].next=edgeHead[v];
    edge[k].re=k - 1;
    edgeHead[v]=k ++;
}

bool spfa(){                //寻找最长增广路时如果是正向边则+相应的费用,反向边-相应的费用
    int i, head = 0, tail = 1;    //  长注释的地方就是从最小费用改到最大费用时需要变动的地方
    for(i = 0; i <= n; i ++){
        dis[i] = inf;////////////
        vis[i] = false;
    }
    dis[0] = 0;
    que[0] = 0;
    vis[0] = true;
    while(head != tail){
        int u = que[head];
        for(i = edgeHead[u]; i != 0; i = edge[i].next){
            int v = edge[i].v;
            if(edge[i].cap && dis[v] >dis[u] + edge[i].cost){////////
                dis[v] = dis[u] + edge[i].cost;
                pre[v] = i;                        // 这个pre数组记录的是从边号为i的那条边去往v
                if(!vis[v]){
                    vis[v] = true;
                    que[tail ++] = v;
                    if(tail == 490) tail = 0;             // 这里用到了循环队列,节省空间
                }
            }
        }
        vis[u] = false;
        head++;
        if(head ==490) head = 0;
    }
    if(dis[n] ==inf) return false;///////////
    return true;
}

void end(){
    int u, p,mi;
	mi=inf;
	for(u = n; u != 0; u = edge[edge[p].re].v){
        p = pre[u];
        if(edge[p].cap<mi)
            mi=edge[p].cap;
    }
    for(u = n; u != 0; u = edge[edge[p].re].v){
        p = pre[u];
        edge[p].cap -= mi;
        edge[edge[p].re].cap += mi;
        ans += edge[p].cost*mi;
    }
}
int main(){
	int i,j,T,t,v,lim,a,b,w;
	scanf("%d",&T);
	for(t=1;t<=T;t++){
	    k=1;
		memset(edgeHead,0,sizeof(edgeHead));
		scanf("%d %d",&v,&lim);
		for(i=1;i<=v;i++){
			scanf("%d %d %d",&a,&b,&w);
			li[i*2-1].be=a;
			li[i*2-1].num=-i;
			li[i*2].be=b;
			li[i*2].num=i;

			line[i].a=a;
			line[i].b=b;
			line[i].w=w;
		}
		qsort(&li[1],2*v,sizeof(li[1]),cmp);
		int temp=li[1].be,tp=1;
		for(i=1;i<=v*2;i++){
			if(li[i].be!=temp){
				tp++;
				temp=li[i].be;
			}
			if(li[i].num>0) //注意这里不能写成if(li[i].num)因为大于0和小于0都满足........在这里T了无数次
				line[li[i].num].b=tp;
			else
				line[-li[i].num].a=tp;
		}
		for(i=0;i<=tp;i++){
			addEdge(i,i+1,lim,0);
		}
		n=tp+1;
		for(i=1;i<=v;i++){
			addEdge(line[i].a,line[i].b,1,-line[i].w);
		}
	    ans=0;
		while(spfa())end();
		printf("%d\n",-ans);
	}
    return 0;
}


 

用STL更简捷

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int inf=99999999;
struct{
    int v, cap, cost, next, re;
}edge[2005];
struct{
	int a,b,w;
}line[210];
int li[450];
int n,ans;
int k,edgeHead[500];
int que[500],pre[500],dis[500];
bool vis[500];

void addEdge(int u,int v,int ca,int co){
    edge[k].v=v;
    edge[k].cap=ca;
    edge[k].cost=co;
    edge[k].next=edgeHead[u];
    edge[k].re=k + 1;               //这个用来记录此边的反边
    edgeHead[u]=k ++;
    edge[k].v=u;
    edge[k].cap=0;
    edge[k].cost=-co;               //故这里去反费用(因为是反向边)
    edge[k].next=edgeHead[v];
    edge[k].re=k - 1;
    edgeHead[v]=k ++;
}

bool spfa(){                //寻找最长增广路时如果是正向边则+相应的费用,反向边-相应的费用
    int i, head = 0, tail = 1;    //  长注释的地方就是从最小费用改到最大费用时需要变动的地方
    for(i = 0; i <= n; i ++){
        dis[i] = inf;////////////
        vis[i] = false;
    }
    dis[0] = 0;
    que[0] = 0;
    vis[0] = true;
    while(head != tail){
        int u = que[head];
        for(i = edgeHead[u]; i != 0; i = edge[i].next){
            int v = edge[i].v;
            if(edge[i].cap && dis[v] >dis[u] + edge[i].cost){////////
                dis[v] = dis[u] + edge[i].cost;
                pre[v] = i;                        // 这个pre数组记录的是从边号为i的那条边去往v
                if(!vis[v]){
                    vis[v] = true;
                    que[tail ++] = v;
                    if(tail == 490) tail = 0;             // 这里用到了循环队列,节省空间
                }
            }
        }
        vis[u] = false;
        head++;
        if(head ==490) head = 0;
    }
    if(dis[n] ==inf) return false;///////////
    return true;
}

void end(){
    int u, p,mi;
	mi=inf;
	for(u = n; u != 0; u = edge[edge[p].re].v){
        p = pre[u];
        if(edge[p].cap<mi)
            mi=edge[p].cap;
    }
    for(u = n; u != 0; u = edge[edge[p].re].v){
        p = pre[u];
        edge[p].cap -= mi;
        edge[edge[p].re].cap += mi;
        ans += edge[p].cost*mi;
    }
}
int main(){
	int i,j,T,t,v,lim,a,b,w;
	int cnt;
	scanf("%d",&T);
	for(t=1;t<=T;t++){
	    k=1;
	    cnt=0;
		memset(edgeHead,0,sizeof(edgeHead));
		scanf("%d %d",&v,&lim);
		for(i=1;i<=v;i++){
			scanf("%d %d %d",&a,&b,&w);
			li[cnt++]=a;
			li[cnt++]=b;

			line[i].a=a;
			line[i].b=b;
			line[i].w=w;
		}
		sort(li,li+cnt);
		n=unique(li,li+cnt)-li;
		for(i=1;i<=n;i++)
            addEdge(i-1,i,lim,0);
		for(i=1;i<=v;i++){
		    int x=lower_bound(li,li+n,line[i].a)-li+1;
		    int y=lower_bound(li,li+n,line[i].b)-li+1;
			addEdge(x,y,1,-line[i].w);
		}
	    ans=0;
		while(spfa())end();
		printf("%d\n",-ans);
	}
    return 0;
}



 

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:486369次
    • 积分:8404
    • 等级:
    • 排名:第2315名
    • 原创:359篇
    • 转载:28篇
    • 译文:0篇
    • 评论:56条
    最新评论