Bzoj - 4552 排序 [二分答案] [线段树]

4552: [Tjoi2016&Heoi2016]排序

Time Limit: 60 Sec   Memory Limit: 256 MB
Submit: 1400   Solved: 710
[ Submit][ Status][ Discuss]

Description

在2016年，佳媛姐姐喜欢上了数字序列。因而他经常研究关于序列的一些奇奇怪怪的问题，现在他在研究一个难题

，需要你来帮助他。这个难题是这样子的：给出一个1到n的全排列，现在对这个全排列序列进行m次局部排序，排

序分为两种：1:(0,l,r)表示将区间[l,r]的数字升序排序2:(1,l,r)表示将区间[l,r]的数字降序排序最后询问第q

位置上的数字。

Input

输入数据的第一行为两个整数n和m。n表示序列的长度，m表示局部排序的次数。1 <= n, m <= 10^5第二行为n个整

数，表示1到n的一个全排列。接下来输入m行，每一行有三个整数op, l, r, op为0代表升序排序，op为1代表降序

排序, l, r 表示排序的区间。最后输入一个整数q，q表示排序完之后询问的位置, 1 <= q <= n。1 <= n <= 10^5

，1 <= m <= 10^5

Output

 输出数据仅有一行，一个整数，表示按照顺序将全部的部分排序结束后第q位置上的数字。

Sample Input

6 3
1 6 2 5 3 4
0 1 4
1 3 6
0 2 4
3

Sample Output

5

HINT

Source

二分p上的数（因为是个排列所以不担心重复），将小于mid的数设为0，大于mid的数设为一，按照信息排序，假如最后p上是0，则小了，L = mid + 1，否则R = mid

复杂度 O(nlogn ^ 2)（吐槽BZOJ cout 会WA）

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5 + 137;

#define ls bt << 1
#define rs bt << 1 | 1
int n, m, a[N], flag[N << 2], sum[N << 2], Mid, qry[N][3], q;

void Build( int bt, int lf, int rg )
{
	flag[bt] = -1;
	if(lf == rg)
	{
		sum[bt] = a[lf] >= Mid + 1;
		return;
	}
	int mid = (lf + rg) >> 1;
	Build(ls, lf, mid);
	Build(rs, mid + 1, rg);
	sum[bt] = sum[ls] + sum[rs];
}

void Push( int bt, int lf, int rg )
{
	if(flag[bt] != -1)
	{
		int mid = (lf + rg) >> 1;
		sum[ls] = (mid - lf + 1) * flag[bt];
		sum[rs] = (rg - mid) * flag[bt];
		flag[ls] = flag[rs] = flag[bt];
		flag[bt] = -1;
	}
}

int Query( int bt, int lf, int rg, int L, int R )
{
//	printf( "%d %d %d %d %d\n", lf, rg, L, R, sum[bt] );
	if(L <= lf && rg <= R) return sum[bt];
	int mid = (lf + rg) >> 1, ans = 0;
	Push(bt, lf, rg);
	if(L <= mid) ans += Query(ls, lf, mid, L, R);
	if(R > mid) ans += Query(rs, mid + 1, rg, L, R);
	sum[bt] = sum[ls] + sum[rs];
	return ans;
}

void Modify( int bt, int lf, int rg, int L, int R, int del )
{
//	printf( "%d %d %d %d\n", lf, rg, L, R );
	if(L <= lf && rg <= R)
	{
		flag[bt] = del;
		sum[bt] = (rg - lf + 1) * del;
		return;
	}
	int mid = (lf + rg) >> 1;
	Push(bt, lf, rg);
	if(L <= mid) Modify(ls, lf, mid, L, R, del);
	if(R > mid) Modify(rs, mid + 1, rg, L, R, del);
	sum[bt] = sum[ls] + sum[rs];
}

int Half_Ans( int L, int R )
{
	while(L < R)
	{
		Mid = (L + R) >> 1;
		Build(1, 1, n);
		for(int i = 1; i <= m; ++i)
		{
			int cnt = Query(1, 1, n, qry[i][1], qry[i][2]);
			if(!cnt) continue;
			if(qry[i][0] == 0)
			{
				Modify(1, 1, n, qry[i][2] - cnt + 1, qry[i][2], 1);
				Modify(1, 1, n, qry[i][1], qry[i][2] - cnt, 0);
			}
			else
			{
				Modify(1, 1, n, qry[i][1], qry[i][1] + cnt - 1, 1);
				Modify(1, 1, n, qry[i][1] + cnt, qry[i][2], 0);
			}
		}
		if(Query(1, 1, n, q, q) == 0) R = Mid;
		else L = Mid + 1;
	}
	return L;
}

int main()
{
	cin >> n >> m;
	for(int i = 1; i <= n; ++i) scanf( "%d", &a[i] );
	for(int i = 1; i <= m; ++i)
		scanf( "%d%d%d", &qry[i][0], &qry[i][1], &qry[i][2] );
	cin >> q;
	printf( "%d\n", Half_Ans(1, n) );
	return 0;
}
/*
5 3
1 2 3 4 5
1 1 3
0 2 4
1 3 5
4

5 3
5 4 3 2 1
1 1 3
0 2 4
1 3 5
1
*/