【luogu P3608 [USACO17JAN]Balanced Photo平衡的照片】 题解

题目链接:https://www.luogu.org/problemnew/show/P3608

乍一看很容易想到O(N^2)的暴力。

对于每个H[i]从ii-1找L[i]再从i+1n找R[i],然后比较。

60分(数据够水)

但是这个思路就是很直白的模拟,让人不容易想到如何去优化。

然后我们换一个也是差不多O(N^2)的思路:

我们设法把H[i]所对应的第k大的k求出来。

for example:

H 34 6 23 0 5 99 2
Kth 2 4 3 7 5 1 6

那么我们就能发现,对于第一个思路从ii-1找一遍再从i+1n找一遍这样两遍的遍历可以变成:

从1~i遍历,计数比当前Kth[i]小的作为L[i],那么R[i] = Kth[i] - L[i] - 1.

这个思路是有优化空间的。

到目前为止,我们需要的是快速的求出L[i]。

设想我们有一个初始全为0的长度为n的序列,从Kthx开始每次在这个序列的第Kth[x]的位置上加1,这样维护到当前第i个Kth时,前i-1Kth是维护过的,如果有比当前第Kth[i]小的就一定会提前处理过(在序列位置上加1)。

于是我们保证了答案的顺序不会出现问题。

就可以采用树状数组维护:

变成一个单点修改,查询0~i的和的问题。

code:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e5 + 10;
struct cow{
	int val, pos;
}a[maxn];
bool cmp(cow a, cow b)
{
	return a.val > b.val;
}
int n, answer, b[maxn];
class BIT{
	public:
		int tree[maxn<<2];
		void update(int pos, int val)
		{
			while(pos <= n)
			{
				tree[pos] += val;
				pos += lowbit(pos);
			}
		}
		int query(int pos)
		{
			int res = 0;
			while(pos)
			{
				res += tree[pos];
				pos -= lowbit(pos);
			}
			return res;
		}
	private:
		int lowbit(int x)
		{
			return x & -x;
		}
}T;
int main()
{
	scanf("%d",&n);
	for(int i = 1; i <= n; i++)
	{
		scanf("%d",&a[i].val);
		a[i].pos = i;
	}
	sort(a + 1, a + 1 + n, cmp);
	for(int i = 1; i <= n; i++) b[a[i].pos] = i;
	for(int i = 1; i <= n; i++)
	{
		int L = T.query(b[i]);
		int R = b[i] - L - 1;
		if(L < R) swap(L, R);
		if(L > R + R) answer++;
		T.update(b[i], 1);
	}
	printf("%d",answer);
}

后记:

小trick:

求一个数列的Kth方法。

posted @ 2019-06-11 21:47  Misaka_Azusa  阅读(190)  评论(0编辑  收藏
Live2D