标签：
题目链接：BZOJ - 1112
枚举每一个长度为k的连续区间，求出这个区间的最优答案，更新全局答案。
可以发现，这个区间的所有柱子最终都变成这k个数的中位数时最优，那么我们就需要查询这个区间的中位数了。
找到中位数之后，我们还应该求出这个区间内小于中位数的数的和，大于中位数的数的和，从而求出操作步数。
这些需要求的值可以用线段树或平衡树来写，我写的是线段树，但是实际上这是一道POI的题目，在MAIN上的空间限制只有35MB，线段树应该是不行的。
因为平衡树只需要 O(n) 空间，所以平衡树才是正解。
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <cstdio>
using namespace std;
const int MaxN = 100000 + 5, MaxNode = 100000 * 20 + 15, MN = 1000000 + 5;
typedef long long LL;
int n, k, Index, Root;
int A[MaxN], T[MaxNode], Son[MaxNode][2];
const LL INF = 999999999999;
LL Ans;
LL Sum[MaxNode];
inline LL gmin(LL a, LL b) {return a < b ? a : b;}
inline void Read(int &Num)
{
char c; c = getchar();
while (c < ‘0‘ || c > ‘9‘) c = getchar();
Num = c - ‘0‘; c = getchar();
while (c >= ‘0‘ && c <= ‘9‘)
{
Num = Num * 10 + c - ‘0‘;
c = getchar();
}
}
void Add(int &x, int s, int t, int Pos, int Num)
{
if (x == 0) x = ++Index;
T[x] += Num;
Sum[x] += (LL)Pos * (LL)Num;
if (s == t) return;
int m = (s + t) >> 1;
if (Pos <= m) Add(Son[x][0], s, m, Pos, Num);
else Add(Son[x][1], m + 1, t, Pos, Num);
}
int Kth(int x, int s, int t, int k)
{
if (s == t) return s;
int ret, m = (s + t) >> 1;
if (T[Son[x][0]] >= k) ret = Kth(Son[x][0], s, m, k);
else ret = Kth(Son[x][1], m + 1, t, k - T[Son[x][0]]);
return ret;
}
LL GetSum(int x, int s, int t, int l, int r)
{
if (l <= s && r >= t) return Sum[x];
int m = (s + t) >> 1;
LL ret = 0ll;
if (l <= m && Son[x][0]) ret += GetSum(Son[x][0], s, m, l, r);
if (r >= m + 1 && Son[x][1]) ret += GetSum(Son[x][1], m + 1, t, l, r);
return ret;
}
int GetNum(int x, int s, int t, int l, int r)
{
if (l <= s && r >= t) return T[x];
int m = (s + t) >> 1;
int ret = 0;
if (l <= m && Son[x][0]) ret += GetNum(Son[x][0], s, m, l, r);
if (r >= m + 1 && Son[x][1]) ret += GetNum(Son[x][1], m + 1, t, l, r);
return ret;
}
int main()
{
scanf("%d%d", &n, &k);
for (int i = 1; i <= n; ++i) Read(A[i]);
Root = Index = 0;
A[0] = 0;
for (int i = 0; i <= k - 1; ++i) Add(Root, 0, MN, A[i], 1);
Ans = INF;
int t = k / 2 + 1, Temp;
LL Now;
for (int i = k; i <= n; ++i)
{
Add(Root, 0, MN, A[i - k], -1);
Add(Root, 0, MN, A[i], 1);
Temp = Kth(Root, 0, MN, t);
Now = (LL)GetNum(Root, 0, MN, 0, Temp - 1) * (LL)Temp - GetSum(Root, 0, MN, 0, Temp - 1);
Now += GetSum(Root, 0, MN, Temp + 1, MN) - (LL)GetNum(Root, 0, MN, Temp + 1, MN) * (LL)Temp;
Ans = gmin(Ans, Now);
}
printf("%lld\n", Ans);
return 0;
}
[BZOJ 1112] [POI2008] 砖块Klo 【区间K大】
标签：
原文地址：http://www.cnblogs.com/JoeFan/p/4324577.html
