题目描述
给定一个数轴上的 n 个区间,要求在数轴上选取最少的点使得第 i 个区间 [ai, bi] 里至少有 ci 个点
使用差分约束系统的解法解决这道题。
Input
输入第一行一个整数 n 表示区间的个数,接下来的 n 行,每一行两个用空格隔开的整数 a,b 表示区间的左右端点。1 <= n <= 50000, 0 <= ai <= bi <= 50000 并且 1 <= ci <= bi - ai+1。
Output
输出一个整数表示最少选取的点的个数
样例
Input
5
3 7 3
8 10 3
6 8 1
1 3 1
10 11 1
Output
6
思路分析
本题要求通过差分约束来进行区间选点,转化为图论中的单源最短路的问题。要求最小解,即要跑最长路,答案为sum[max]。
Sum[i]表示[0,i]之间选点个数,那[a, b] 里至少有 c 个点即可以看作sum[b+1]-sum[a]>=c。
且要使sum有意义,必须保证0<=sum[i]-sum[i-1]<=1;
AC代码
#include<iostream>
#include<queue>
#include<cstring>
#include<cmath>
const int inf=1e8;
using namespace std;
int head[50005],sum[50005],inq[50005];
int tot;
struct Edge
{
int v,w,next;//相邻的点,权值,下条边的序号
} ;
Edge edge[200010];
void init()
{
//初始化
tot=0;
memset(head,-1,sizeof(head));
memset(inq,0,sizeof(inq));
}
void add(int u,int v,int w)
{
edge[tot].v=v;
edge[tot].w=w;
edge[tot].next=head[u];
head[u]=tot;
tot++;
}
void spfa(int s)
{
queue<int> q;
sum[s] = 0;
q.push(s);
inq[s] = 1;
while (!q.empty())
{
int u = q.front(); q.pop();
inq[u] = 0;
for (int i= head[u];i!=-1;i= edge[i].next)
{
int v=edge[i].v;
if(sum[v]<sum[u]+edge[i].w)
{
sum[v]=sum[u]+edge[i].w;
if (!inq[v])
{
q.push(v);
inq[v]=1;
}
}
}
}
return;
}
int main()
{
ios::sync_with_stdio(false);
init();
int n;
cin>>n;
int a,b,c,Min=50005,Max=0;
for(int i=0;i<n;i++)
{
cin>>a>>b>>c;
add(a,b+1,c);
Min=min(Min,a);
Max=max(Max,b+1);//当前覆盖区间
}
for(int i=0;i<=Max;i++)
{
sum[i]=-inf;
}
for(int i=1;i<=Max;i++)//0<=sum[i+1]-sum[i]<=1
{
add(i-1,i,0);
add(i,i-1,-1);
}
spfa(0);
cout<<sum[Max]<<endl;
return 0;
}