bzoj 4867: [Ynoi2017]舌尖上的由乃 分块

题意

给一个序列，要求资瓷区间加和求区间第k小。
n,m<=100000

分析

用数据结构显然不好做，考虑分块。
首先对于每一块都维护一个有序的数组，数组第一维表示权值，第二维表示编号。
对于修改操作，整块的我们可以打标记。对于多余的块，我们可以将其分成要修改和不修改两部分，修改完后再归并排序回去就可以仍然保持其有序了。
对于询问操作，可以先二分答案，然后在每一块上二分。注意多余的块不能每次暴力扫，应该第一次先把两边的块提出来，然后在上面二分。
这样的话块的大小取$\sqrt nlogn$是最优的，总的复杂度是$O(n\sqrt nlogn)$

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

const int N=100005;

int n,m,len,tim,dfn[N],w[N],t1[N],t2[N],block,last[N],cnt,mx[N],sta[N],end[N],tag[N],dep[N],bel[N];
struct edge{int to,next,w;}e[N];
pair<int,int> b[N],p1[N],p2[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void addedge(int u,int v,int w)
{
    e[++cnt].to=v;e[cnt].w=w;e[cnt].next=last[u];last[u]=cnt;
}

void dfs(int x)
{
    dfn[x]=++tim;w[tim]=dep[x];
    for (int i=last[x];i;i=e[i].next)
        dep[e[i].to]=dep[x]+e[i].w,dfs(e[i].to);
    mx[x]=tim;
}

void prepare()
{
    dfs(1);
    block=sqrt(n)*log(n)/log(2);
    for (int i=1;i<=n;i++)
    {
        bel[i]=(i+block-1)/block;
        end[bel[i]]=i;
        if (!sta[bel[i]]) sta[bel[i]]=i;
        b[i]=make_pair(w[i],i);
    }
    for (int i=1;i<=bel[n];i++) sort(b+sta[i],b+end[i]+1);
}

int binary(int l,int r,pair<int,int> u)
{
    int L=l,R=r;
    while (L<=R)
    {
        int mid=(L+R)/2;
        if (b[mid]>u) R=mid-1;
        else L=mid+1;
    }
    return R-l+1;
}

int query(int l,int r,int k)
{
    if (k>r-l+1) return -1;
    if (bel[l]==bel[r])
    {
        int tot=0;
        for (int i=sta[bel[l]];i<=end[bel[l]];i++)
            if (b[i].second>=l&&b[i].second<=r) p1[++tot]=b[i];
        return p1[k].first+tag[bel[l]];
    }
    int b1=0,b2=0;
    for (int i=sta[bel[l]];i<=end[bel[l]];i++)
        if (b[i].second>=l) p1[++b1]=b[i];
    for (int i=sta[bel[r]];i<=end[bel[r]];i++)
        if (b[i].second<=r) p2[++b2]=b[i];
    int L=0,R=2000000;
    while (L<=R)
    {
        int mid=(L+R)/2;
        int s=0;pair<int,int> u;
        u=make_pair(mid-tag[bel[l]],n+1);s+=upper_bound(p1+1,p1+b1+1,u)-p1-1;
        u=make_pair(mid-tag[bel[r]],n+1);s+=upper_bound(p2+1,p2+b2+1,u)-p2-1;
        for (int i=bel[l]+1;i<bel[r];i++) u=make_pair(mid-tag[i],n+1),s+=binary(sta[i],end[i],u);
        if (s>=k) R=mid-1;
        else L=mid+1;
    }
    return R+1;
}

void merge(int l,int r,int w)
{
    int b1=0,b2=0;
    for (int i=sta[bel[l]];i<=end[bel[l]];i++)
        if (b[i].second>=l&&b[i].second<=r) p1[++b1]=b[i];
        else p2[++b2]=b[i];
    for (int i=1;i<=b1;i++) p1[i].first+=w;
    int i=sta[bel[l]],j=1,k=1;
    while (j<=b1||k<=b2)
        if (k>b2||k<=b2&&j<=b1&&p1[j].first<p2[k].first) b[i++]=p1[j],j++;
        else b[i++]=p2[k],k++;
}

void modify(int l,int r,int w)
{
    if (bel[l]==bel[r]) {merge(l,r,w);return;}
    merge(l,end[bel[l]],w);merge(sta[bel[r]],r,w);
    for (int i=bel[l]+1;i<bel[r];i++) tag[i]+=w;
}

int main()
{
    n=read();m=read();len=read();
    for (int i=2,x,w;i<=n;i++) x=read(),w=read(),addedge(x,i,w);
    prepare();
    while (m--)
    {
        int op=read(),x=read(),k=read();
        if (op==1) printf("%d\n",query(dfn[x],mx[x],k));
        else modify(dfn[x],mx[x],k);
    }
    return 0;
}