POJ - 1201 Intervals 差分约束 || 贪心 + 线段树

传送门：POJ 1201

题意：给出n个三元组（ai, bi, ci）,找出一组元素数目最少的整数集合，使得集合中包含在[ai, bi]之间的数不少于ci个。

思路：设d[i]表示[0..i]区间内选中了几个整数，则题给条件可以转化为d[bi] - d[ai - 1] >= ci，并且由于d[i]是一个前缀和函数，还满足 0  <= d[i] - d[i - 1] <= 1， 同时d[-1] = 0,题目转化为解不等式方程组，用差分约束求解。

代码：

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<queue>
#define ll long long
#define pb push_back
#define fi first
#define se second
#define pi acos(-1)
#define inf 0x3f3f3f3f
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define rep(i,x,n) for(int i=x;i<n;i++)
#define per(i,n,x) for(int i=n;i>=x;i--)
using namespace std;
typedef pair<int,int>P;
const int MAXN=100010;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int pre[MAXN << 1];
int cnt;
struct node{
	int v, w;
	int next;
}mp[MAXN << 1];
void init()
{
	cnt = 0;
	memset(pre, -1, sizeof(pre));
}
void add(int u, int v, int w)
{
	mp[cnt].v = v;
	mp[cnt].w = w;
	mp[cnt].next = pre[u];
	pre[u] = cnt++;
}
queue<int> q;
bool book[MAXN];
int dis[MAXN];
void spfa(int s)
{
	int u, v, w;
	q.push(s);
	book[s] = 1;
	memset(dis, -inf, sizeof(dis));
	dis[s] = 0;
	while(!q.empty())
	{
		u = q.front(); q.pop();
		for(int i = pre[u]; i != -1; i = mp[i].next)
		{
			v = mp[i].v, w = mp[i].w;
			if(dis[v] < dis[u] + w)
			{
				dis[v] = dis[u] + w;
				if(!book[v])
				{
					book[v] = 1;
					q.push(v);
				}
			}
		}
		book[u] = 0;
	}
}
int main()
{
	init();
	int n;
	cin >> n;
	int down = inf, up = 0, u, v, w;
	for(int i = 0; i < n; i++)
	{
		scanf("%d %d %d", &u, &v, &w);
		add(u, v + 1, w);
		up = max(up, v + 1);
		down = min(down, u);
	}
	for(int i = down + 1; i <= up; i++)
	add(i - 1, i, 0), add(i, i - 1, -1);
	spfa(down);
	cout << dis[up] << endl;
 	return 0;
}

这个题还可以贪心 + 线段树求解， 将区间排序后，贪心的在每个区间最后选数，并用线段树记录选了哪几个数字，求解下一个区间的时候，先询问线段树该区间内已经选了几个数字。