题目描述
给定一个数轴上的 n 个区间,要求在数轴上选取最少的点使得第 i 个区间 [ai, bi] 里至少有 ci 个点。
使用差分约束系统的解法解决这道题!
输入输出
Input
输入第一行一个整数 n 表示区间的个数,接下来的 n 行,每一行两个用空格隔开的整数 a,b 表示区间的左右端点。1 <= n <=
50000, 0 <= ai <= bi <= 50000 并且 1 <= ci <= bi - ai+1。
Output
输出一个整数表示最少选取的点的个数
Sample Input
5
3 7 3
8 10 3
6 8 1
1 3 1
10 11 1
Sample Output
6
思路分析
此题与前几周做的区间选点有所不同,那一题要求每个区间至少有一个点,用贪心算法解决。对于该题,要求第i个区间至少有ci个点,学长给出了新的解法——差分约束。
差分约束类似于最短路中的松弛操作,对于不等式约束xi-xj<=ck,可以转化成dis[i]<=dis[j]+w(i,j),移项得dis[j]>=dis[i]+w(i,j)(边权可正可负可为零),因此可以转化为图论中的最短路问题求解。
这道题目首先需要构造不等式组:
记sum[i]表示数轴上[0,i]之间选点的个数
对于第i个区间[ai,bi]需要满足sum[bi]-sum[ai-1]>=ci
然后根据构造关系,该题是>=,因此跑最长路解决
注意&总结
1.本题边权可以为负,因此不能用Dijkstra解决,这里用SPFA求最长路。
2.关于memset的用法,虽然见到学长这样用感觉很方便,就用了几次,但不甚明白原理,出错了,这里总结一下。
memset不是对每个int赋值
!而是单个字节逐个copy
!
如memset(a,0,sizeof(a)) 一般一个int占4个字节,赋值之后是00000000 00000000 00000000 00000000
而memset(a,1,sizeof(a))就变成了00000001 00000001 00000001 00000001一个int并不是十进制的“1”
赋值为-1也是可以的,补码存储。
若想初始化为一个很大的数,赋值127就可以了;
赋值128就会初始化为一个很小的数10000000 10000000 10000000 10000000
大约是-2e9。
代码
#include<iostream>
#include<stdio.h>
#include<queue>
#include<string.h>
using namespace std;
int n,a,b,c,tot;
const int maxn=5e4+5,size=1e6+5;
int mina=maxn,maxb=0;
int sum[maxn],inq[maxn],head[maxn];
int inf=-1e9;//跑最长路inf初始化为最小值
queue<int> q;
struct edge{
int to,w,nxt;
}e[size];//注意边数组的大小
void addEdge(int x,int y,int w)
{//前向星
e[++tot].to=y;
e[tot].w=w;
e[tot].nxt=head[x];
head[x]=tot;
}
void spfa(int s)
{
while(!q.empty()) q.pop();
//queue<int> q;
sum[s]=0,inq[s]=1;
q.push(s);
while(!q.empty())
{
int u=q.front();
q.pop();
inq[u]=0;
for(int i=head[u];i!=0;i=e[i].nxt)
{
int v=e[i].to;
if(sum[v]<sum[u]+e[i].w)
{//最长路
sum[v]=sum[u]+e[i].w;
if(!inq[v])
{
q.push(v);
inq[v]=1;
}
}
}
}
}
void init()
{
tot=0;
for(int i=0;i<maxn;i++)
{//初始化时记得考虑是仅初始化数组中存数的部分还是全部
head[i]=0;
sum[i]=inf;
inq[i]=0;
}
}
int main()
{
scanf("%d",&n);
init();
for(int i=0;i<n;i++)
{
scanf("%d %d %d",&a,&b,&c);
addEdge(a-1,b,c);
mina=min(mina,a);
maxb=max(maxb,b);
}
for(int i=mina;i<=maxb;i++)
{
addEdge(i-1,i,0);
addEdge(i,i-1,-1);
}
spfa(mina-1);
printf("%d\n",sum[maxb]);
return 0;
}