专题四——线段树、树状数组

树状数组：快速求前缀和 O(logn)
两种基本操作：
①给某个位置上的数加上一个数
②求某一个前缀和

1.1264. 动态求连续区间和【三个模板要记住】

#include <bits/stdc++.h>

using namespace std;
const int N=100010;
int n,m;
int a[N],tr[N];
int lowbit(int x)
{
    return x&-x;
}
void add(int x,int v)
{
    for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=v;
}
int query(int x)
{
    int res=0;
    for(int i=x;i;i-=lowbit(i)) res+=tr[i];
    return res;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) add(i,a[i]);
    while(m--)
    {
        int k,x,y;
        scanf("%d%d%d",&k,&x,&y);
        if(k==0) printf("%d\n",query(y)-query(x-1));
        else add(x,y);
    }
    return 0;
}

2.1265. 数星星

#include <bits/stdc++.h>

using namespace std;
const int N=32010;
int n;
int tr[N],level[N];
int lowbit(int x)//2^k，x的二进制表示后有k个0
{
    return x&-x;
}
void add(int x,int v)//在某一个横坐标下+1
{
    for(int i=x;i<N;i+=lowbit(i)) tr[i]+=v;
}
int sum(int x)//相当于前缀和数组里的s[N]
{
    int res=0;
    for(int i=x;i;i-=lowbit(i)) res+=tr[i];
    return res;
}
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        x++;
        level[sum(x)]++;
        add(x,1);
    }
    for(int i=0;i<n;i++) printf("%d\n",level[i]);
    return 0;
}

线段树：实现单点修改和区间查询
①pushup:用子节点信息更新当前节点信息
②build:在一段区间上初始化线段树
③modify:修改
④query:查询
x为树的节点下标；父节点：x/2——>x>>1;左鹅子2x——>x<<1;右鹅子2x+1——>x<<1|1

1264. 动态求连续区间和【线段树做法】

#include <bits/stdc++.h>

using namespace std;
const int N=100010;
int n,m;
int w[N];
struct Node
{
    int l,r;
    int sum;
}tr[N*4];
void pushup(int u)//用子节点信息更新当前节点信息
{
    tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}
void build(int u,int l,int r)
{
    if(l==r) tr[u]={l,r,w[r]};
    else
    {
        tr[u]={l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid),build(u<<1|1,mid+1,r);//递归
        pushup(u);//回溯
    }
}
//查询区间始终不变，询问的是当前结点左右边界是否被完全包裹
int query(int u,int l,int r)
{
    if(tr[u].l>=l&&tr[u].r<=r) return tr[u].sum;
    int mid=tr[u].l+tr[u].r>>1;
    int sum=0;
    if(l<=mid ) sum=query(u<<1,l,r);
    if(r>mid) sum+=query(u<<1|1,l,r);
    return sum;
}
void modify(int u,int x,int v)
{
    if(tr[u].l==tr[u].r) tr[u].sum+=v;
    else
    {
        int mid=tr[u].l+tr[u].r>>1;
        if(x<=mid) modify(u<<1,x,v);
        else modify(u<<1|1,x,v);
        pushup(u);
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    build(1,1,n);
    int k,a,b;
    while(m--)
    {
        scanf("%d%d%d",&k,&a,&b);
        if(k==0) printf("%d\n",query(1,a,b));
        else modify(1,a,b);
    }
    return 0;
}

3.1270. 数列区间最大值

与上面的模板完全类似，只是pushup函数稍有变动

#include <bits/stdc++.h>

using namespace std;
const int N=100010;
int n,m;
int w[N];
struct Node
{
    int l,r;
    int maxv;
}tr[N*4];
void build(int u,int l,int r)
{
    if(l==r) tr[u]={l,r,w[r]};
    else
    {
        tr[u]={l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid);
        build(u<<1|1,mid+1,r);
        tr[u].maxv=max(tr[u<<1].maxv,tr[u<<1|1].maxv);
    }
}
int query(int u,int l,int r)
{
    if(tr[u].l>=l&&tr[u].r<=r) return tr[u].maxv;
    int mid=tr[u].l+tr[u].r>>1;
    int maxv=INT_MIN;
    if(l<=mid) maxv=query(u<<1,l,r);
    if(r>mid) maxv=max(maxv,query(u<<1|1,l,r));
    return maxv;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    build(1,1,n);
    int l,r;
    while(m--)
    {
        scanf("%d%d",&l,&r);
        printf("%d\n",query(1,l,r));
    }
    return 0;
}

4.1215小朋友排队

【冒泡排序与逆序对数量】

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
const int N=1000010;
int n;
int h[N],tr[N];
int sum[N];
int lowbit(int x)
{
    return x&-x;
}
//相当于桶排序的前缀和操作
void add(int x,int v)
{
    for(int i=x;i<N;i+=lowbit(i)) tr[i]+=v;
}
int query(int x)
{
    int res=0;
    for(int i=x;i;i-=lowbit(i)) res+=tr[i];
    return res;
}
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++) {scanf("%d",&h[i]);h[i]++;}
    //求每个数前面有多少个数比它大
    for(int i=0;i<n;i++)
    {
        sum[i]=query(N-1)-query(h[i]);
        add(h[i],1);
    }
    memset(tr,0,sizeof tr);
    //求每个数后面有多少个数比它小
    for(int i=n-1;i>=0;i--)
    {
        sum[i]+=query(h[i]-1);
        add(h[i],1);
    }
    //高斯求和
    LL res=0;
    for(int i=0;i<n;i++) res+=(LL)sum[i]*(sum[i]+1)/2;
    cout<<res<<endl;
    return 0;
}

5.1237. 螺旋折线

找规律类的题（i了i了）：首先发现四个顶点的特殊性，其次判断目标点在哪一条边上，最后加上偏移量

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
int main()
{
    int x,y;
    cin>>x>>y;
    if(abs(x)<=y)//在上方
    {
        int n=y;
        cout<<(ll)(2*n-1)*(2*n)+x-(-n)<<endl;
    }
    else if(abs(y)<=x)
    {
        int n=x;
        cout<<(ll)(2*n)*(2*n)+n-y<<endl;
    }
    else if(abs(x)<=abs(y)+1&&y<0)
    {
        int n=abs(y);
        cout<<(ll)(2*n)*(2*n+1)+n-x<<endl;
    }
    else
    {
        int n=abs(x);
        cout<<(ll)(2*n-1)*(2*n-1)+y-(-n+1)<<endl;
    }
    return 0;
}