【洛谷】AT_abc371_d [ABC371D] 1D Country 的题解

【洛谷】AT_abc371_d [ABC371D] 1D Country 的题解

洛谷传送门

AT传送门

题解

锐评：比C题还简单

题目大意：给定你 $n$ 个地址，在第 $x[i]$ 位置有 $p[i]$ 个人，给定 $q$ 个询问，每次查询位置 $l$ 到 $r$之间有多少个人。

首先，惊人的数据范围让我直接放弃了暴力。考虑线性算法，刚开始和大多数人一样考虑到了前缀和。但是后面看了题解想到了二分。唯一的问题就是负数的问题，这可以将 $l$,$r$,进行一个映射，让它成为一个在 $1$ 到 $n$ 之间的数。但是我很懒，直接用 upper_bound 和 lower_bound 水过去了qaq。

Caution：不开 long long 见祖宗！！！

代码

#include <bits/stdc++.h>
#define lowbit(x) x & (-x)
#define endl "\n"
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
namespace fastIO {
	inline int read() {
		register int x = 0, f = 1;
		register char c = getchar();
		while (c < '0' || c > '9') {
			if(c == '-') f = -1;
			c = getchar();
		}
		while (c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
		return x * f;
	}
	inline void write(int x) {
		if(x < 0) putchar('-'), x = -x;
		if(x > 9) write(x / 10);
		putchar(x % 10 + '0');
		return;
	}
}
using namespace fastIO;
int n, q, x[200005], p[200005], ans[200005];
int main() {
	//freopen(".in","r",stdin);
	//freopen(".out","w",stdout);
	n = read();
	for(int i = 1; i <= n; i ++) {
		x[i] = read();
	}
	for(int i = 1; i <= n; i ++) {
		p[i] = read();
		ans[i] = ans[i - 1] + p[i];
	}
	q = read();
	while(q --) {
		int l, r;
		l = read(), r = read();
		l = lower_bound(x + 1, x + n + 1, l) - x;
		r = upper_bound(x + 1, x + n + 1, r) - x;
		write(ans[r - 1] - ans[l - 1]), putchar('\n');
	}
	return 0;
}