BZOJ 1112 POI2008 砖块

最新推荐文章于 2018-10-22 20:25:41 发布

Fuxey

最新推荐文章于 2018-10-22 20:25:41 发布

阅读量940

点赞数 2

分类专栏： POI 树状数组文章标签： POI 树状数组

本文链接：https://blog.csdn.net/Fuxey/article/details/50649893

版权

POI 同时被 2 个专栏收录

3 篇文章 0 订阅

订阅专栏

树状数组

2 篇文章 0 订阅

订阅专栏

树状数组， Treap , 还有MultiSet&Map 都是可搞的。
这里提供一个树状数组的版本……

提示：
1. 对于这种点间求距离的题目，初中课本上讲中位数应该提到过……

代码后详解：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <string>
#include <vector>
#include <deque>
#include <stack>
#include <algorithm>
#include <set>

using namespace std;
typedef long long ll;

const int maxn = 1010000;

__inline int re() {
    int n = 0, ch = getchar(); bool flag = false;
    while(!isdigit(ch)) flag |= ch == '-', ch = getchar();
    while(isdigit(ch)) n = n * 10 + ch - '0', ch = getchar();
    return flag ? -n : n;
}

int n , k;
int a[maxn] , Max;

ll d1[1100000] , d2[1100000];

int lowbit(int x) { return x& -x; }
void add(int x , int a , int n , ll* d)
{
    while(x<=n)
    {
        d[x]+= a;
        x+= lowbit(x);
    }
}

ll query(int x , int n , ll* d)
{
    ll res =0;
    while(x>0)
    {
        res+= d[x];
        x-= lowbit(x);
    }
    return res;
}

int findK(int k , ll* d)
{
    int res = 0 , cnt = 0;
    for(int i=20;i>=0;i--)
    {
        res+= 1<<i;
        if(res>Max || cnt+d[res]>=k) res-= 1<<i;
        else cnt+= d[res];
    }
    return res+1;
}

int main(int argc, char *argv[]) {

    n = re(); k = re();
    for(int i=1;i<=n;i++) a[i] = re() , Max =max(Max , a[i]);
    for(int i=1;i<=n+1;i++) a[i]++;

    int cnt = k;
    ll res = 1e15 , sum = 0;
    for(int i=1;i<=k;i++) add(a[i], 1, Max, d1) , add(a[i], a[i], Max, d2) , sum+= a[i];
    do
    {
        int now = findK((k+1)/2, d1);
        ll  m1 = query(now, Max, d1);
        ll  m2 = query(now, Max, d2);
        res = min(res , sum-2*m2+m1*now-(k-m1)*now);

        sum-= a[cnt-k+1];
        add(a[cnt-k+1], -1, Max, d1);
        add(a[cnt-k+1], -a[cnt-k+1], Max, d2);

        sum+= a[cnt+1];
        add(a[cnt+1], 1, Max, d1);
        add(a[cnt+1], a[cnt+1], Max, d2);
        cnt++;
    }while(cnt<=n);

    printf("%lld\n" , res);
    return 0;
}