bzoj4843

4843: [Neerc2016]Expect to Wait

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 29   Solved: 17
[ Submit][ Status][ Discuss]

Description

ls最近开了一家图书馆，大家听说是ls开的，纷纷过来借书，自然就会出现供不应求的情况， 并且借书的过程类

似一个队列，每次有人来借书就将它加至队尾，每次有人来还书就把书借给队头的若干个人，定义每个人的等待时

间为拿到书的时刻减去加至队列的时刻，如果一个人根本就拿不到书，则等待时间为inf，现在给出所有时刻借书

还书的情况，和若干个询问，每次询问当图书馆初始有x本书时所有人的等待时间之和是多少（如果存在一个人根

本拿不到书，则输出INFINITY）。

Input

第一行两个整数n,q（1<=n,q<=100000),表示有n个时刻有借书还书的情况，以及有q个询问。

接下来n行，每行表示一个操作，操作如下：

1."+ t k" 在t时刻有k本书被还回来。

2."- t k" 在t时刻有k个人来借书。

（1<=t<=1e9,1<=k<=10000）

输入顺序保证t递增。

接下来一行q个数，第i个数bi（1<=bi<=1e9）表示图书馆初始有bi本书，询问所有人的等待时间之和为多少。

Output

一共q行，每行一个数表示等待时间之和，如果存在一个人根本拿不到书，则输出INFINITY。

Sample Input

5 4
- 1 1
- 2 2
+ 4 1
- 6 1
+ 7 2
0 3 1 2

Sample Output

INFINITY
0
8
3

HINT

题解：

考虑每个时刻有多少人等待，求和即为答案。

还书记为+1，借书记为-1，

每次询问就是求一段序列每个数扣掉x后小于0的元素的和（要把时刻离散化成时间段）。

把询问离线排序，每个数比x大之后永远比x大，所以可以用线段树维护区间小于0的和，和最大的有意义的数是多少，询问后将序列每个数加上两个询问之差。

修改后暴力把大于0的数修改为无意义的，每个数会被暴力改一次，所以时间复杂度是对的。

因为每次询问修改都是整个数列，也可以排序后单调地扫，复杂度一样，但是常数会小，可能还好写。

但是我的代码并没有变快QAQ。

#include <stdio.h>
#include <algorithm>

using namespace std;

//long long
const long long maxn = 120000;
struct so {
	long long x , i;
} s[maxn];
struct so1 {
	long long x , ran;
} s1[maxn];
long long n , q;
long long tim[maxn] , a[maxn] , ran[maxn];
char z[10];
long long now , all;
long long last;
long long ans[maxn];
bool cmp ( so x1 , so x2 ) {
	return x1.x < x2.x;
}
bool cmp1 ( so1 x1 , so1 x2 ) {
	return x1.x < x2.x;
}
void work () {
	long long i , j;
	scanf ( "%lld%lld" , &n , &q );
	for ( i = 1 ; i <= n ; i++ ) {
		scanf ( "%s%lld%lld" , z + 1 , &tim[i] , &a[i] );
		if ( z[1] == '-' ) a[i] *= -1;
	}
	tim[n+1] = tim[n] + 1;
	for ( i = 1 ; i <= n ; i++ ) s1[i].ran = tim[i+1] - tim[i];
	for ( i = 1 ; i <= n ; i++ ) {
		a[i] += a[i-1];
		s1[i].x = a[i];
	}
	for ( i = 1 ; i <= q ; i++ ) {
		scanf ( "%lld" , &s[i].x );
		s[i].i = i;
	}
	last = a[n];
	sort ( s + 1 , s + 1 + q , cmp );
	sort ( s1 + 1 , s1 + 1 + n , cmp1 );
	for ( j = 1 ; j <= n ; j++ ) {
		if ( s1[j].x < 0 ) {
			//printf ( "%d\n" , s1[j].x );
			now += s1[j].x * s1[j].ran;
			all += s1[j].ran;
		}
		else {
			break;
		}
	}
	j--;
	for ( i = 1 ; i <= q ; i++ ) {
		if ( s[i].x + last < 0 ) {
			ans[s[i].i] = -1;
			continue;
		}
		while ( j >= 1 && s1[j].x + s[i].x >= 0 ) {
			now -= s1[j].x * s1[j].ran;
			all -= s1[j].ran;
			j--;
		}
		//printf ( "%lld %lld %lld %lld \n" , i , j , now , all );
		ans[s[i].i] = -now - s[i].x * all;
	}
	for ( i = 1 ; i <= q ; i++ ) {
		if ( ans[i] == -1 ) printf ( "INFINITY\n" );
		else printf ( "%lld\n" , ans[i] );
	}
}
int main () {
	//FILE *fpr = freopen ( "bzoj4843.in" , "r" , stdin );
	//FILE *fpw = freopen ( "bzoj4843.out" , "w" , stdout );
	work ();
	return 0;
}