Week8 作业A 区间选点 差分约束

题目描述

给定一个数轴上的 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;
}