差分约束解决区间选点(week8-A)

参考这个

至于为什么最小解是找寻最长路，最大解找最小路，看这篇博客

区间选点题目

给定一个数轴上的 n 个区间，要求在数轴上选取最少的点使得第 i 个区间 [ai, bi] 里至少有 ci 个点

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

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

5
3 7 3
8 10 3
6 8 1
1 3 1
10 11 1

6

解题思路

区间选点除了贪心算法，还可以用差分约束（结合最短路径SPFA求解）。

代码

#include<iostream>
#include<stdio.h>
#include<queue>
#include<stdlib.h>
using namespace std;

const int maxn=55000;//最大点数 
const int maxm=600100;//最大边数 
const int inf=1e9;
queue<int> Q;
//前向星存储图
//图中边、点的编号从0开始 
struct edge
{
	int v,w,next;
}edges[maxm];
int head[maxn],dis[maxn],inq[maxn];
int tot=0;
//dis存储【0，i】的已选的点的个数 ,cnt[i]记录路径中经过的边数 
//int n;//图中的个数 
void init()
{
	tot=0;
	for(int i=0;i<maxn;i++)
		head[i]=-1;//初始区间内没有选点 
}
void addEdge(int _u,int _v,int _w)
{
	edges[tot].v=_v;
	edges[tot].w=_w;
	edges[tot].next=head[_u];
	head[_u]=tot;
	tot++;
 } 
void initData()
{
	while(!Q.empty())Q.pop();
	for(int i=0;i<maxn;i++)
		dis[i]=-inf,inq[i]=0;
}
void spfa(int s)
{
	initData();
	inq[s]=1,dis[s]=0;
	Q.push(s);
	while(!Q.empty())
	{
		int u=Q.front();Q.pop();
		inq[u]=0;
		for(int i=head[u];i!=-1;i=edges[i].next)
		{
			
			int v=edges[i].v;
			if(dis[v]<dis[u]+edges[i].w)
			{
				dis[v]=dis[u]+edges[i].w;
				if(!inq[v])
				{
					inq[v]=1;
					Q.push(v);
				}
			}
		}
	}
}




int main()
{
	int n,nn,mina;
	scanf("%d",&n);
	int a,b,c;
	nn=-1,mina=inf;
	init();
	while(n--)
	{
		scanf("%d %d %d",&a,&b,&c);
		//因为含有区间0，所以将各点向右移一个单位  
		//变为：dis[b]-dis[a-1]>=c
		//即在图中，加入权重为c的有向边a->b+1 
		addEdge(a,b+1,c);//图中点的编号依旧从0开始，但dis[0]==0
		if(b>nn)nn=b; 
		if(a<mina)mina=a;
	}
	nn+=2;//图中点的个数 
	//整数选点0<=dis[i]-dis[i-1]<=1,即dis[i]>=dis[i-1]+0,dis[i-1]>=dis[i]+(-1) 
	for(int i=mina+1;i<nn;i++)
	{
		addEdge(i-1,i,0);
		addEdge(i,i-1,-1);
	}
	//用spfa求距离原点0最长路获得最小解
	spfa(mina);
	printf("%d\n",dis[nn-1]);
	//for(int i=0;i<nn;i++)
	//	printf("%d",dis[i]);
	
	return 0;
 }