HYSBZ3110-K大数查询

[Zjoi2013]K大数查询

Time Limit: 20 Sec Memory Limit: 512 MB
Submit: 9350 Solved: 2761
[ Submit][ Status][ Discuss]

Description

有N个位置，M个操作。操作有两种，每次操作如果是1 a b c的形式表示在第a个位置到第b个位置，每个位置加入一个数c
如果是2 a b c形式，表示询问从第a个位置到第b个位置，第C大的数是多少。

Input

第一行N，M
接下来M行，每行形如1 a b c或2 a b c

Output

输出每个询问的结果

Sample Input

2 5
1 1 2 1
1 1 2 2
2 1 1 2
2 1 1 1
2 1 2 3

Sample Output

1
2
1

HINT

【样例说明】

第一个操作后位置 1 的数只有 1 ，位置 2 的数也只有 1 。第二个操作后位置 1

的数有 1 、 2 ，位置 2 的数也有 1 、 2 。第三次询问位置 1 到位置 1 第 2 大的数是

1 。第四次询问位置 1 到位置 1 第 1 大的数是 2 。第五次询问位置 1 到位置 2 第 3

大的数是 1 。‍

N,M<=50000,N,M<=50000

a<=b<=N

1操作中abs(c)<=N

2操作中c<=Maxlongint

Source

解题思路：线段树套线段树，外面一棵是权值线段树，里面一棵是区间线段树

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cctype>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <vector>
#include <bitset>
#include <functional>using namespace std;#define LL long long
const int INF = 0x3f3f3f3f;
const int maxn = 50009;struct node
{int op, l, r;long long z;
}a[maxn];
long long x[maxn];
int n, m, cnt, tot;
int s[200005], L[20000005], R[20000005];
long long sum[20000005], lazy[20000005];void Push(int k, int l, int r)
{if (!L[k]) L[k] = ++tot;if (!R[k]) R[k] = ++tot;lazy[L[k]] += lazy[k]; lazy[R[k]] += lazy[k];int mid = (l + r) >> 1;sum[L[k]] += 1LL * (mid - l + 1)*lazy[k];sum[R[k]] += 1LL * (r - mid)*lazy[k];lazy[k] = 0;
}void update(int &k, int l, int r, int ll, int rr)
{if (!k) k = ++tot;if (ll <= l&&rr >= r){sum[k] += 1LL * (r - l + 1);lazy[k]++;return;}if(lazy[k]) Push(k, l, r);int mid = (l + r) >> 1;if (ll <= mid) update(L[k], l, mid, ll, rr);if (rr > mid) update(R[k], mid + 1, r, ll, rr);sum[k] = sum[L[k]] + sum[R[k]];
}void Insert(int ll, int rr, int p)
{int k = 1, l = 1, r = cnt - 1;while (l != r){int mid = (l + r) >> 1;update(s[k], 1, n, ll, rr);if (p <= mid) r = mid, k = k << 1;else l = mid + 1, k = k << 1 | 1;}update(s[k], 1, n, ll, rr);
}LL query(int k, int l, int r, int ll, int rr)
{if (!k) return 0;if (ll <= l&&rr >= r) return sum[k];int mid = (l + r) >> 1;LL ans = 0;if(lazy[k]) Push(k, l, r);if (ll <= mid) ans += query(L[k], l, mid, ll, rr);if (rr > mid) ans += query(R[k], mid + 1, r, ll, rr);return ans;
}int solve(int ll, int rr, LL p)
{int l = 1, r = cnt - 1, k = 1;while (l != r){int mid = (l + r) >> 1;LL temp = query(s[k << 1 | 1], 1, n, ll, rr);if (temp >= p) l = mid + 1, k = k << 1 | 1;else r = mid, k <<= 1, p -= temp;}return l;
}int main()
{scanf("%d%d", &n, &m);for (int i = 1; i <= m; i++) scanf("%d%d%d%lld", &a[i].op, &a[i].l, &a[i].r, &a[i].z), x[i] = a[i].z;sort(x + 1, x + 1 + m);cnt = unique(x + 1, x + 1 + m) - x;for (int i = 1; i <= m; i++){if (a[i].op == 1){int k = lower_bound(x + 1, x + cnt, a[i].z) - x;Insert(a[i].l, a[i].r, k);}else printf("%lld\n", x[solve(a[i].l, a[i].r, a[i].z)]);}return 0;
}