丽娃河的狼人传说
Time limit per test: 1.0 seconds
Time limit all tests: 1.0 seconds
Memory limit: 256 megabytes
丽娃河是华师大著名的风景线。但由于学校财政紧缺,丽娃河边的路灯年久失修,一到晚上就会出现走在河边要打着手电的情况,不仅非常不方便,而且影响安全:已经发生了大大小小的事故多起。
方便起见,丽娃河可以看成是从 1 到 n 的一条数轴。为了美观,路灯只能安装在整数点上,每个整数点只能安装一盏路灯。经专业勘测,有 m 个区间特别容易发生事故,所以至少要安装一定数量的路灯,
请问至少还要安装多少路灯。
Input
第一行一个整数 T (1≤T≤300),表示测试数据组数。
对于每组数据:
第一行三个整数 n,m,k (1≤n≤103,1≤m≤103,1≤k≤n)。
第二行 k 个不同的整数用空格隔开,表示这些位置一开始就有路灯。
接下来 m 行表示约束条件。第 i 行三个整数 li,ri,ti 表示:第 i 个区间 [li,ri] 至少要安装 ti 盏路灯 (1≤li≤ri≤n,1≤ti≤n)。
Output
对于每组数据,输出 Case x: y。其中 x 表示测试数据编号(从 1 开始),y 表示至少要安装的路灯数目。如果无解,y 为 −1。
思路:对每个区间查询是不是满足要求,不满足要求就要加灯。可以很容易想到是个单点更新区间查询和。。。然后就写了个树状数组。。。不足加灯是个单点更新,然后处理之前是查询。。两个注意的点。。排序是按r的升序排。。这考虑到对后面的贡献,来神还是强2333我一开始是按l升序后来感觉不对。第二点是你在插灯的时候要先把已经有的灯数去了然后在没有灯的地方插灯。。想了想我连这里都错了 还是蠢啊。。
不过听czh大神说就直接一个一个加起来也能对2333
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<vector>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<string>
#include<stack>
#include<map>
using namespace std;
//thanks to pyf ...
#define INF 0x3f3f3f3f
#define CLR(x,y) memset(x,y,sizeof(x))
#define mp(x,y) make_pair(x,y)
typedef pair<int, int> PII;
typedef long long ll;
const int N = 1e6 + 5;
struct Seg
{
int l, r;
int need;
} s[N];
int n, m, k;
int vis[N];
int a[N];
bool cmp(Seg a, Seg b)
{
return a.r < b.r || (a.r == b.r && a.l > b.l);
}
int lowbit(int x)
{
return x & (-x);
}
void update(int pos, int val)
{
while (pos <= n)
{
a[pos] += val;
pos += lowbit(pos);
}
}
int query(int pos)
{
int res = 0;
while (pos > 0)
{
res += a[pos];
pos -= lowbit(pos);
}
return res;
}
int main()
{
int T;
scanf("%d", &T);
int ka = 0;
while (T--)
{
CLR(vis, 0);
CLR(a, 0);
CLR(s, 0);
scanf("%d%d%d", &n, &m, &k);
for (int i = 0; i < k; i++)
{
int pos;
scanf("%d", &pos);
vis[pos] = 1;
update(pos, 1);
}
for (int i = 0; i < m; i++)
scanf("%d%d%d", &s[i].l, &s[i].r, &s[i].need);
sort(s, s + m, cmp);
int ans = 0;
int flag = 0;
for (int i = 0; i < m; i++)
{
int l = s[i].l, r = s[i].r, need = s[i].need;
if (query(r) - query(l - 1) >= need)
continue;
else if (r - l + 1 < need)
{
flag = 1;
break;
}
else
{
need -= (query(r) - query(l - 1));
for (int j = r; j >= l; j--)
{
if (!vis[j])
{
vis[j] = 1;
need -- ;
ans ++;
update(j, 1);
}
if (!need)
break;
}
}
}
if (flag)
printf("Case %d: -1\n", ++ka);
else
printf("Case %d: %d\n", ++ka, ans);
}
}