【BZOJ4209】西瓜王

Description

西瓜王xgw最近喜欢上了玩星际争霸，很神的他尤其喜欢玩神族。有一天他生产了非常多的高阶圣堂和黑暗圣堂，并把他们排成了一条直线。
但是他觉得这些东西奇形怪状的很不好看，于是他想把他们全都变成末日执政官和黑暗执政官那样的圆圆的像西瓜一样的东西。但是他的人口上限不够了……他只能从中选出一些来变成执政官。
但是他玩的星际争霸有点特别，每一个圣堂有一个x值x_i，若x_i为奇数则他为高阶圣堂，否则为黑暗圣堂。而x值越大会使得合成以后的执政官更圆（TAT实在没啥好说的了）。xgw自然是想使得x值的和最大啦。
xgw毕竟也是一个王，所以他很任性。他每次会在屏幕上随便框一下（也就是搞一个区间），然后问你这个区间要合成k个执政官选出的圣堂的x值之和最大值是多少。
p.s.两个高阶圣堂可以合成一个末日执政官，两个黑暗圣堂可以合成一个黑暗执政官
Input

第一行一个整数n表示圣堂的个数。
第二行n个整数表示每个圣堂的x值。
第三行一个整数T表示xgw框了多少次。
接下来T行每行三个整数l,r,k代表要从[l, r]这段区间里选出k个圣堂去合成k/2个执政官，保证k是偶数。
Output

输出T行，对于xgw每框一次输出选出的圣堂的x值之和最大值。若没有可行的选择方案则输出-1

Sample Input

5

7 5 3 4 2

1

1 5 4
Sample Output

18
HINT

N,M<=3*10^5,Xi<=10^9

Source

想法和PA2014Final Bazarek相似,同样是考虑较大数奇换偶,偶换奇
因为要区间询问k大的奇数/偶数,所以可以考虑用整体二分/主席树

这里选择了主席树(练板子)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 300010
#define SIZE 6000000
#define GET (ch>='0'&&ch<='9')
#define LL long long
using namespace std;
int n,m,c[2],cnt;
int a[MAXN],b[MAXN];
int root[MAXN],ls[SIZE],rs[SIZE],val[SIZE][2];
LL sum[SIZE],Sum,ans;
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
inline void insert(int l,int r,int x,int &y,int pos,int v)
{
    val[y=++cnt][0]=val[x][0];val[y][1]=val[x][1];
    sum[y]=sum[x]+v;val[y][v&1]++;
    if (l==r)   return;
    int mid=(l+r)>>1;ls[y]=ls[x];rs[y]=rs[x];
    if (pos<=mid)   insert(l,mid,ls[x],ls[y],pos,v);
    else    insert(mid+1,r,rs[x],rs[y],pos,v);
} 
inline int kth(int l,int r,int x,int y,int k,int o)
{
    if (l==r)   return b[l];
    int mid=(l+r)>>1;
    int tmp=val[rs[y]][o]-val[rs[x]][o];
    if (k>tmp)  return kth(l,mid,ls[x],ls[y],k-tmp,o);
    else    return kth(mid+1,r,rs[x],rs[y],k,o);
    return 0;
}
inline void query(int l,int r,int x,int y,int k)
{
    if (l==r)   {   c[b[l]&1]+=k;Sum+=1ll*b[l]*k;return;    }
    int mid=(l+r)>>1;
    int tmp=val[rs[y]][0]+val[rs[y]][1]-val[rs[x]][0]-val[rs[x]][1];
    if (k>tmp)  
        c[0]+=val[rs[y]][0]-val[rs[x]][0],c[1]+=val[rs[y]][1]-val[rs[x]][1],
        Sum+=sum[rs[y]]-sum[rs[x]],query(l,mid,ls[x],ls[y],k-tmp);
    else    query(mid+1,r,rs[x],rs[y],k);
}
int main()
{
    in(n);int l,r,k;
    for (int i=1;i<=n;i++)  in(a[i]),b[i]=a[i];
    sort(b+1,b+n+1);
    for (int i=1;i<=n;i++)  insert(1,n,root[i-1],root[i],lower_bound(b+1,b+n+1,a[i])-b,a[i]);
    for (in(m);m;m--)
    {
        in(l);in(r);in(k);
        c[0]=c[1]=Sum=0;ans=-1;
        if (k>r-l+1)    {   puts("-1");continue;    }
        query(1,n,root[l-1],root[r],k);
        if (!(c[0]&1))  {   printf("%lld\n",Sum);continue;  }
        for (int i=0;i<=1;i++)
            if (c[i]&&(c[i^1]<val[root[r]][i^1]-val[root[l-1]][i^1]))
                ans=max(ans,Sum-kth(1,n,root[l-1],root[r],c[i],i)+kth(1,n,root[l-1],root[r],c[i^1]+1,i^1));
        printf("%lld\n",ans);   
    }
}