Codeforces 387E George and Cards(二分+树状数组)

题目链接:Codeforces 387E George and Cards


题目大意:给出一个长度为n的序列,然后在给出一个长度为k的条件序列,要求将原先的删除n-k个后得到的序列满足条件序列,条件序列b[i],表示序列的前i个元素中必须包含有b[i]这个数。删除操作,选取一个区间,删除中间的最小值,并且获得相应的区间长度的什么奖励,问说最多可以获得多少奖励(奖励啥看不懂)。


解题思路:二分+树状数组。首先,可以分析的出结论,b数组即为原先数组删除n-k个后的目标数组(想一下就知道的b[1]的位置肯定是放值为b[1]的卡片,那么值为b[2]的卡片就只能放在2上,以次类推);其次,为了得到尽量多的奖励,在删除元素时选取的区间就要尽量长,不仅如此,在删除元素时也要从小的开始删除,因为删除大的会导致删除小的时候区间长度的缩短;这样的话就可以通过将序列每个数按照从小到大的顺序将位置放入set中,碰到需要删除的元素就去二分上下限。知道上下限就可以通过BIT求得中间存在元素的个数。


#include <stdio.h>
#include <string.h>
#include <iostream>
#include <set>

using namespace std;

const int N = 1000005;
typedef long long ll;

int n, k, p[N], b[N];
ll v[N];

void add(int x, ll val) {
	while (x <= n) {
		v[x] += val;
		x += (x & (-x));
	}
}

ll sum(int x) {
	ll ans = 0;
	while (x > 0) {
		ans += v[x];
		x -= (x & (-x));
	}
	return ans;
}

void init() {
	int a;
	memset(b, 0, sizeof(b));
	memset(v, 0, sizeof(v));
	scanf("%d%d", &n, &k);

	for (int i = 1; i <= n; i++) {
		scanf("%d", &a);
		p[a] = i;
		add(i, 1);
	}

	for (int i = 1; i <= k; i++) {
		scanf("%d", &a);
		b[a] = 1;
	}
}

ll solve() {
	set<int> s;
	set<int>::iterator iter;
	ll ans = 0;

	s.insert(0); s.insert(n+1);
	for (int i = 1; i <= n; i++) {
		if (b[i]) {
			s.insert(p[i]);
			continue;
		}
		iter = s.upper_bound(p[i]);
		int r = *iter - 1;
		int l = *(--iter);
		ans += sum(r) - sum(l);
		add(p[i], -1);
	}
	return ans;
}

int main() {
	init();
	cout << solve() << endl;
	return 0;
}


Codeforces 387E George and Cards(二分+树状数组)

上一篇:TXT阅读神器 分分钟打造一本电子书


下一篇:Ubuntu安装google-chrome浏览器