GYM 101755 G.Underpalindromity（BIT）

Description

给出一个长度为$n$的序列，对于这个序列的$n-k+1$个长度为$k$的子序列，定义每个子序列的权值为把该子序列变成回文序列的最小代价，每次可以花费一代价把该序列的任一元素加一，问这$n-k+1$个子序列的权值和

Input

第一行两个整数$n,k$，之后输入$n$个整数$a_1,...,a_n$$(1\le k\le n\le 2\cdot 10^5,-10^8\le a_i\le 10^8)$

Output

输出这$n-k+1$个子序列的权值和

Sample Input

3 2
3 1 2

Sample Output

3

Solution

对于$a_i$，如果$a_j$在某个长度为$k$的子序列与$a_i$处于对称位置，那么这一对数对答案的贡献即为$|a_i-a_j|$，现在考虑 两个问题，第一个问题：可以与$a_i$配对的$a_j$的范围；第二个问题：$a_i$对答案的贡献

对于可以与$a_i$配对的$a_j$的范围，如果$i\ge k$，那么最左边可以与$i$配对的显然是$a_{i-k+1}$，否则考虑在第一个长度为$k$的子序列$a_1,...,a_k$中与$a_i$配对的$a_{k+1-i}$，如果$i\le n-k+1$，那么最右边可以与$i$配对的显然是$a_{i+k-1}$，否则考虑在第$n-k+1$个长度为$k$的子序列$a_{n-k+1},...,a_n$中与$a_i$配对的$a_{2n-k+1-i}$，以此求出了可以与$a_i$配对的$a_j$即为满足$L\le j\le R$且$j$与$L$奇偶性相同（显然$a_j$每次下标增加$2$），其中$L,R$即为上面的分析中最左边和最右边可以与$a_i$配对的元素下标

对于$a_i$的贡献，共有$s=\frac{R-L}{2}+1$个$a_j$可以与$a_i$配对，其中有若干值小于$a_i$，有若干值大于$a_i$，假设有$x$个$j$使得$a_j<a_i$，$y$个$j$使得$a_j>a_i$，那么$a_i$对答案的贡献即为$(x-y)\cdot a_i$，故问题只剩下求出$x,y$了，简单分析可知，$k$为奇数时，是奇数$j$对奇数$i$有贡献，偶数$j$对偶数$i$有贡献，$k$为偶数时相反，那么我们分别维护下标为奇数的值和下标为偶数的值，把$a_i$升序排，一个个插到树状数组中（用两个树状数组分别维护下标为奇数和偶数的值），在树状数组的$i$位置加一，对于$a_i$，首先把所有值严格小于$a_i$的值插到树状数组中，然后求出与$a_i$可以配对的$a_j$的合法区间$[L,R]$，根据$j$的奇偶找到对应的树状数组查询该区间和即为这个区间中严格小于$a_i$的$a_j$个数$x$，然后把所有值等于$a_i$的值插进去，再查询一次即得到区间$[L,R]$中小于等于$a_i$的$a_j$个数$x'$，进而$y=s-x'$，$a_i$对答案的贡献即为$(x-y)\cdot a_i=(x+x'-s)\cdot a_i$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int maxn=200005;
int n,m,k,ans[maxn];
struct BIT 
{
    #define lowbit(x) (x&(-x))
    int b[maxn],n;
    void init(int _n)
    {
        n=_n;
        for(int i=1;i<=n;i++)b[i]=0;
    }
    void update(int x,int v)
    {
        while(x<=n)
        {
            b[x]+=v;
            x+=lowbit(x);
        }
    }
    int query(int x)
    {
        int ans=0;
        while(x)
        {
            ans+=b[x];
            x-=lowbit(x);
        }
        return ans;
    }
}bit[2];
int get(int i,int &L,int &R)
{
    L=i-k+1,R=i+k-1;
    if(L<1)L=2-L;
    if(R>n)R=2*n-R;
}
#define val first
#define id second
P a[maxn];
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)scanf("%d",&a[i].val),a[i].id=i;
    sort(a+1,a+n+1);
    int L,R;
    bit[0].init(n),bit[1].init(n);
    for(int i=1;i<=n;)
    {
        int j=i;
        while(j+1<=n&&a[j+1].val==a[i].val)j++;
        for(int l=i;l<=j;l++)
        {
            get(a[l].id,L,R);
            ans[a[l].id]=bit[(k+a[l].id+1)&1].query(R)-bit[(k+a[l].id+1)&1].query(L-1);
        }
        for(int l=i;l<=j;l++)bit[a[l].id&1].update(a[l].id,1);
        for(int l=i;l<=j;l++)
        {
            get(a[l].id,L,R);
            ans[a[l].id]+=(bit[(k+a[l].id+1)&1].query(R)-bit[(k+a[l].id+1)&1].query(L-1)-((R-L)/2+1));
        }
        i=j+1;
    }
    ll sum=0;
    for(int i=1;i<=n;i++)
        sum+=(ll)a[i].val*ans[a[i].id];
    printf("%lld\n",sum);
    return 0;
}