hdu 5213

问题描述

wld有n个数(a1...an)

保证对于任意$1 \leq i \leq n，1 \leq ai \leq n$

wld有一个常数k保证2 \leq k\leq 2*n2≤k≤2∗n

为了消除歧义保证k为奇数

他有mm个询问

每个询问有参数l1,r1,l2,r2l1,r1,l2,r2

保证(1 \leq l1\leq r1 < l2 \leq r2 \leq n)(1≤l1≤r1<l2≤r2≤n)

对于每个询问你需要回答有多少个二元组$(i，j)$满足：

l1 \leq i\leq r1l1≤i≤r1且l2 \leq j \leq r2l2≤j≤r2且ai + aj = kai+aj=k

保证1 \leq n\leq 30000, 1 \leq m \leq 300001≤n≤30000,1≤m≤30000

输入描述

多组数据(最多55组)

对于每组数据：

第一行：一个数nn表示数的个数

接下来一行：一个数kk表示wld的常数

接下来一行：nn个数，依次为$a1, a2, … an$

接下来一行：一个数mm表示询问数

接下来mm行：四个数l1, r1, l2, r2l1,r1,l2,r2表示这组询问的参数

输出描述

对于每组数据：

对于每个询问输出二元组的数目

输入样例

5

3

1 2 1 2 3

1

1 2 3 5

输出样例

2

Hint

a1 + a4 = 3

a2 + a3 = 3

思路：序列莫队加上容斥定理

设f(x,y)为区间[x,y]满足a[x]+a[y]=k的个数，则f(l1,r1,l2,r2)=f(l1,r2)-f(r1+1,r2)-f(l1,l2-1)+f(r1+1,l2-1);

代码如下：

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int maxn = 30005;const int blocksize = 200;struct node{int left, right, id, p;void init(int l1, int r1, int id1, int p1){left = l1, right = r1;id = id1, p = p1;}}query[4 * maxn];bool cmp(node &a, node &b){if (a.left / blocksize != b.left / blocksize) return a.left / blocksize < b.left / blocksize;return  a.right < b.right;}int cnt[2 * maxn], a[maxn], ans[maxn], n, k;int main(){while (scanf("%d", &n) != EOF){int cnt1 = 0;scanf("%d", &k);for (int i = 1;i <= n;i++)scanf("%d", &a[i]);memset(cnt, 0, sizeof(cnt));memset(ans, 0, sizeof(ans));int q;scanf("%d", &q);for (int i = 1;i <= q;i++){int l1, r1, l2, r2;scanf("%d%d%d%d", &l1, &r1, &l2, &r2);query[cnt1++].init(l1, r2, i, 1);query[cnt1++].init(r1 + 1, r2, i, -1);query[cnt1++].init(l1, l2 - 1, i, -1);if (r1 + 1 <= l2 - 1) query[cnt1++].init(r1 + 1, l2 - 1, i, 1);}sort(query, query + cnt1, cmp);int left = 1, right = 0, res = 0;for (int i = 0;i < cnt1;i++){while (left > query[i].left){left--;if (k > a[left] && k - a[left] <= n)res += cnt[k - a[left]];cnt[a[left]]++;}while (right < query[i].right){right++;if (k > a[right] && k - a[right] <= n)res += cnt[k - a[right]];cnt[a[right]]++;}while (left < query[i].left){cnt[a[left]]--;if (k > a[left] && k - a[left] <= n)res -= cnt[k - a[left]];left++;}while (right > query[i].right){cnt[a[right]]--;if (k > a[right] && k - a[right] <= n)res -= cnt[k - a[right]];right--;}ans[query[i].id] += res*query[i].p;}for (int i = 1;i <= q;i++)printf("%d\n", ans[i]);}return 0;}