結果

問題 No.1270 Range Arrange Query
ユーザー leaf_1415leaf_1415
提出日時 2020-10-24 00:30:03
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 2,074 ms / 7,000 ms
コード長 5,271 bytes
コンパイル時間 1,926 ms
コンパイル使用メモリ 108,912 KB
実行使用メモリ 7,248 KB
最終ジャッジ日時 2024-07-21 14:20:16
合計ジャッジ時間 11,975 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
5,248 KB
testcase_01 AC 3 ms
5,376 KB
testcase_02 AC 4 ms
5,376 KB
testcase_03 AC 3 ms
5,376 KB
testcase_04 AC 4 ms
5,376 KB
testcase_05 AC 4 ms
5,376 KB
testcase_06 AC 141 ms
5,376 KB
testcase_07 AC 1,167 ms
6,384 KB
testcase_08 AC 187 ms
5,376 KB
testcase_09 AC 787 ms
6,032 KB
testcase_10 AC 811 ms
6,276 KB
testcase_11 AC 2,074 ms
7,244 KB
testcase_12 AC 2,050 ms
7,248 KB
testcase_13 AC 2,054 ms
7,112 KB
testcase_14 AC 43 ms
7,116 KB
testcase_15 AC 70 ms
7,120 KB
testcase_16 AC 69 ms
7,244 KB
testcase_17 AC 66 ms
7,120 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <cstdlib>
#include <cassert>
#include <vector>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <map>
#include <set>
#include <bitset>
#include <string>
#include <algorithm>
#include <utility>
#define rep(x, s, t) for(llint (x) = (s); (x) <= (t); (x)++)
#define chmin(x, y) (x) = min((x), (y))
#define chmax(x, y) (x) = max((x), (y))
#define all(x) (x).begin(),(x).end()
#define inf 1e18
#define mod 1000000007

using namespace std;

typedef long long llint;
typedef pair<llint, llint> P;

struct BIT{
	int size;
	vector<llint> bit;
	BIT(){size = 0;}
	BIT(int s){
		size = s;
		bit.resize(size+1);
		init();
	}
	void init(){
		for(int i = 1; i <= size; i++) bit[i] = 0;
	}
	llint query(int i){
		llint ret = 0;
		while(i > 0){
			ret += bit[i];
			i -= i&(-i);
		}
		return ret;
	}
	void add(int i, llint x){
		while(i <= size){
			bit[i] += x;
			i += i&(-i);
		}
	}
};


// range-add, range-min query
 
struct LazySegTree{
	int size;
	vector<llint> seg, delay;
	
	LazySegTree(){}
	LazySegTree(int size){
		this->size = size;
		seg.resize(1<<(size+1));
		delay.resize(1<<(size+1));
	}
	
	void init()
	{
		for(int i = 0; i < (1<<(size+1)); i++){
			seg[i] = inf;
			delay[i] = 0;
		}
	}
	
	void eval(int l, int r, int k)
	{
		if(delay[k]){
			seg[k] += delay[k];  //総和クエリのときは区間幅を乗じる必要がある
			if(l < r){
				delay[k*2] += delay[k];
				delay[k*2+1] += delay[k];
			}
			delay[k] = 0;
		}
	}
	
	void update(int i, llint val)
	{
		int l = 0, r = (1<<size)-1, k = 1;
		eval(l, r, k);
		for(int j = size-1; j >= 0; j--){
			k <<= 1;
			if(i & (1<<j)){
				k++;
				r = (l+r)/2;
			}
			else l = (l+r)/2+1;
			eval(l, r, k);
		}
		
		i += (1 << size);
		seg[i] = val;
		while(i > 1){
			i /= 2;
			seg[i] = min(seg[i*2], seg[i*2+1]);
		}
	}
	
	void add(int a, int b, int k, int l, int r, llint val)
	{
		eval(l, r, k);
		
		if(b < l || r < a) return;
		if(a <= l && r <= b){
			delay[k] += val;
			eval(l, r, k);
			return;
		}
		add(a, b, k*2, l, (l+r)/2, val);
		add(a, b, k*2+1, (l+r)/2+1, r, val);
		seg[k] = min(seg[k*2], seg[k*2+1]);
	}
	void add(int a, int b, llint val){
		if(a > b) return;
		add(a, b, 1, 0, (1<<size)-1, val);
	}
 
	llint query(int a, int b, int k, int l, int r)
	{
		eval(l, r, k);
		
		if(b < l || r < a) return inf;
		if(a <= l && r <= b) return seg[k];
		llint lval = query(a, b, k*2, l, (l+r)/2);
		llint rval = query(a, b, k*2+1, (l+r)/2+1, r);
		return min(lval, rval);
	}
	llint query(int a, int b)
	{
		return query(a, b, 1, 0, (1<<size)-1);
	}
};

llint n, Q;
LazySegTree seg(15);
BIT lbit(20005), rbit(20005);

struct Mo{
	typedef pair<P, P> Query;
	int B;
	vector<int> vec;
	vector<Query> qvec;
	vector<P> avec;
	int qid;
	llint sum = 0;
	
	Mo(){
		init();
	}
	void init(){
		vec.clear();
		qvec.clear();
		avec.clear();
		qid = 0;
		
		lbit.init(), rbit.init();
	}
	void setArray(vector<int> &src){
		vec = src;
		B = sqrt((int)src.size());
	}
	int addQuery(int l, int r)
	{
		qvec.push_back(Query(P(l/B, r), P(l, qid)));
		return qid++;
	}
	void calc()
	{
		if(qvec.size() == 0) return;
		sort(qvec.begin(), qvec.end());
		
		int l = qvec[0].second.first, r = qvec[0].first.second;
		for(int i = 0; i < l; i++) erase(vec[i], 0);
		for(int i = (int)vec.size()-1; i > r; i--) erase(vec[i], 1);
		//for(int i = l; i <= r; i++) add(vec[i]);
		avec.push_back(P(qvec[0].second.second, get(l, r)));
		
		for(int i = 1; i < qvec.size(); i++){
			int nl = qvec[i].second.first, nr = qvec[i].first.second;
			while(l < nl) erase(vec[l], 0), l++;
			while(l > nl) l--, add(vec[l], 0);
			while(r < nr) r++, add(vec[r], 1);
			while(r > nr) erase(vec[r], 1), r--;
			int qid = qvec[i].second.second;
			avec.push_back(P(qid, get(l, r)));
		}
		sort(avec.begin(), avec.end());
	}
	int getAnswer(int qid)
	{
		return avec[qid].second;
	}
	
	void add(int x, int l)
	{
		if(l == 0){
			seg.add(1, x-1, -1);
			lbit.add(x, -1);
			sum -= rbit.query(x-1);
		}
		else{
			seg.add(x+1, n, -1);
			rbit.add(x, -1);
			sum -= lbit.query(n) - lbit.query(x);
		}
	}
	void erase(int x, int l)
	{
		if(l == 0){
			seg.add(1, x-1, 1);
			sum += rbit.query(x-1);
			lbit.add(x, 1);
		}
		else{
			seg.add(x+1, n, 1);
			sum += lbit.query(n) - lbit.query(x);
			rbit.add(x, 1);
		}
	}
	int get(llint l, llint r)
	{
		return seg.query(1, n) * (r-l+1) + sum;
	}
};

llint a[20005];
llint l[20005], r[20005], qid[20005];
llint linv[20005], rinv[20005];
BIT bit(20005);
Mo mo;

int main(void)
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	cin >> n >> Q;
	for(int i = 1; i <= n; i++) cin >> a[i];
	
	for(int i = 1; i <= n; i++){
		linv[i] = linv[i-1] + bit.query(n) - bit.query(a[i]);
		bit.add(a[i], 1);
	}
	bit.init();
	for(int i = n; i >= 1; i--){
		rinv[i] = rinv[i+1] + bit.query(a[i]-1);
		bit.add(a[i], 1);
	}
	
	seg.init();
	for(int i = 1; i <= n; i++) seg.update(i, 0);
	
	vector<int> vec;
	for(int i = 1; i <= n; i++) vec.push_back(a[i]);
	mo.setArray(vec);
	
	for(int q = 1; q <= Q; q++){
		cin >> l[q] >> r[q];
		qid[q] = mo.addQuery(l[q]-1, r[q]-1);
	}
	mo.calc();
	
	for(int q = 1; q <= Q; q++){
		llint ans = mo.getAnswer(qid[q]);
		ans += linv[l[q]-1] + rinv[r[q]+1];
		cout << ans << endl;
	}
	
	return 0;
}
0