程序设计Week8——A-区间选点

最新推荐文章于 2022-07-25 21:48:31 发布

李洋233

最新推荐文章于 2022-07-25 21:48:31 发布

阅读量219

点赞数

文章标签：差分法

本文链接：https://blog.csdn.net/qq_43371596/article/details/105578215

版权

A-区间选点

题目描述

给定一个数轴上的 n 个区间，要求在数轴上选取最少的点使得第 i 个区间 [ai, bi] 里至少有 ci 个点，使用差分约束系统完成。

Input

输入第一行一个整数 n 表示区间的个数，接下来的 n 行，每一行两个用空格隔开的整数 a，b 表示区间的左右端点。1 <= n <= 50000， 0 <= ai <= bi <= 50000 并且 1 <= ci <= bi - ai+1。

Output

输出一个整数表示最少选取的点的个数。

解题思路

首先要将区间选点的问题转换成差分约束系统的形式，为此，我们不妨定义dis[i]是0到i上选择的点的个数，这样题目要求的[a,b]区间内至少选择c个点，就可以写成dis[b+1]-dis[a]>=c（因为a是可以为0的，而数组下标不能为0，因此计算时可以将整个区间向右移动1位，整体的计算结果是没有变化的），这样根据差分约束系统到单源最短路的关系来求解；
本题中数轴上的每个点，其实就相当于图的每一个点，而每个区间的选点个数，就相当于图中这两点的边权，由于题目要求是至少为c，因此在图的算法中，这里其实就变成了求单源最长路，而在最长路求解过程中要注意，点是可以重复入队的，因此要及时修改vis的值，而最长路的松弛条件就是dis[v]<dis[u]+w；
另外，为保证每一个dis有意义，需要保证0<=dis[i]-dis[i-1]<=1，这意味着除了题目给定的约束条件（边条件）外，整个图都必须连通，并且正向边权值为0，逆向边权值为-1；
最后，根据所有区间的最大值max，直接输出dis[max+1]（因为移动了1）。

实现代码

#include<iostream>
#include<queue>
using namespace std;

const int num=50020,minn=-1e6;

struct edge
{
	int u,v,w,next;
};
edge E[3*num];			//num是点的个数,同时输入的n满足n<num,边数=n+2*(num-1)<3*num 
int head[num],tot,dis[num];
bool vis[num];

void add(int u,int v,int w)
{
	E[tot].u=u;
	E[tot].v=v;
	E[tot].w=w;
	E[tot].next=head[u];
	head[u]=tot++;
}

void init()
{
	for(int i=0;i<num;++i)
	{
		head[i]=-1;
		vis[i]=false;
		dis[i]=minn;		//在最长路中，这个值要足够小 
	}
	tot=0;
}

void link(int x)
{//用于连通所有的点 
	for(int i=0;i<=x;++i)
	{//正向距离为0，反向距离-1 
		add(i,i+1,0);
		add(i+1,i,-1);
	}
}

void spfa(int p)
{
	queue<int> q;
	dis[p]=0;
	vis[p]=true;
	q.push(p);
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		vis[u]=false;		//求最长路需要重复入队 
		int i=head[u];
		while(i!=-1)
		{
			int v=E[i].v,w=E[i].w;
			if(dis[v]<dis[u]+w)
			{
				dis[v]=dis[u]+w;
				if(!vis[v])
				{
					vis[v]=true;
					q.push(v);
				}
			}
			i=E[i].next;
		}
	}
}

int main()
{
	int n,x=-1;		//x记录所有区间的最右端点值 
	cin>>n;
	init();
	for(int i=0;i<n;++i)
	{
		int a,b,c;
		cin>>a>>b>>c;
		add(a,b+1,c);	//a可以等于0，因此计算时需要讲整个区间向右移动1 
		if(x<b)
			x=b;
	}
	link(x);
	spfa(0);
	cout<<dis[x+1]<<endl;	//因为整体移动了1，所以所求结果应该是x+1的距离 
	return 0;
}