蓝桥杯刷题八

1.动态求连续区间和

两个操作都是树状数组的基本操作 直接写就可以了 不会树状数组的可以翻一下我的文章 其实概念挺简单的

树状数组代码

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int a[N],tr[N];
int n;
int m;
int lowbit(int x)
{
    return x&(-x);
}
void add(int x,int c)
{
    for(int i=x;i<=n;i+=lowbit(i))
        tr[i]+=c;
}
int sum(int x)
{
    int res=0;
    for(int i=x;i;i-=lowbit(i))
        res+=tr[i];
    return res;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    for(int i=1;i<=n;i++)
    {
        add(i,a[i]);
    }
    while(m--)
    {
        int k,a1,b;
        cin>>k>>a1>>b;
        if(k==0)
        {
            cout<<sum(b)-sum(a1-1)<<endl;
        }
        else
        {
            add(a1,b);
        }
    }
    return 0;
}

这题还可以用单点修改的线段树 反正线段树可以解决大部分树状数组的问题

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
struct Node{
    int l,r;
    int sum;
}tr[N*4];
int w[N];
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[l]};
    }
    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(l<=tr[u].l&&tr[u].r<=r)
    {
        return tr[u].sum;
    }
    int mid=tr[u].l+tr[u].r>>1,sum=0;
    if(l<=mid)
    {
        sum+=query(u<<1,l,r);
    }
    if(mid+1<=r)
    {
        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()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>w[i];
    build(1,1,n);
    while(m--)
    {
        int k,a,b;
        cin>>k>>a>>b;
        if(k==0){
            cout<<query(1,a,b)<<endl;
        }
        else
        {
            modify(1,a,b);
        }
    }
    return 0;
}

2.数星星

输入规定了y是升序 那么直接树状数组就可以了

#include <bits/stdc++.h>

using namespace std;

const int N = 32010;

int tr[N], lv[N];

int lowbit(int x) {
    return x & -x;
}

void add(int x) {
    for (int i = x; i < N; i += lowbit(i)) {
        tr[i] ++;
    }
}

int query(int x) {
    int res = 0;
    for (int i = x; i; i -= lowbit(i)) {
        res += tr[i];
    }

    return res;
}

int main() {
    cin.tie(nullptr) -> sync_with_stdio(false);

    int n;
    cin >> n;

    for (int i = 0; i < n; i ++) {
        int x, y;
        cin >> x >> y;
        x++;
        lv[query(x)] ++;
        add(x);
    }

    for (int i = 0; i < n; i ++) {
        cout << lv[i] << "\n";
    }

    return 0;
}

3.数列区间最大值

线段树的模板题哦

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
struct Node{
    int l,r;
    int maxl;
}tr[N*4];
int w[N];
void pushup(int u)
{
    tr[u].maxl=max(tr[u<<1].maxl,tr[u<<1|1].maxl);
}
void build(int u,int l,int r)
{
    if(l==r)
    {
        tr[u]={l,r,w[l]};
    }
    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].maxl;
    int mid=tr[u].l+tr[u].r>>1;
    int maxl=-0x3f3f3f3f;
    if(l<=mid) maxl=query(u<<1,l,r);
    if(r>mid) maxl=max(maxl,query(u<<1|1,l,r));
    return maxl;

}
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>w[i];
    build(1,1,n);
    while(m--)
    {
        int x,y;
        cin>>x>>y;
        cout<<query(1,x,y)<<endl;
    }
    return 0;
}

4.小朋友排队

这题剖析出来就是求 a[i]的前面有多少个比它大 有多少个比它小 这样的话就是求逆序对个数了

做法有两种归并排序和树状数组 树状数组比较方便但是如果元素的大小不受限制的话，就不能用树状数组了 只能归并排序 自己思考一下其中的原因

归并排序代码

#include<iostream>
using namespace std;
const int N = 1e5 + 10;
#define x first
#define y second
typedef pair<int, int> PII;
typedef long long LL;
int n;
PII a[N];  //a[i]={身高，编号}
LL cnt[N];    //cnt[i]表示编号为i的小朋友交换次数
PII temp[N];
void merge(int l, int r)
{
    if (l >= r)  return;
    int mid = l + r >> 1;
    merge(l, mid), merge(mid + 1, r);
    int k = 0, i = l, j = mid + 1;

    while (i <= mid && j <= r)
        if (a[i] <= a[j])
        {
            //相对于i来说,j 前面的数都比它小, j 前面的数都和 i 交换过
            cnt[a[i].y] += j - mid - 1;
            temp[k++] = a[i++];
        }
        else {
            //相对于j来说,i 后面的数都比它大, i 后面的数(包括i)都和 j 交换
            cnt[a[j].y] += mid - i + 1;
            temp[k++] = a[j++];
        }
    while (i <= mid)
    {
        cnt[a[i].y] += j - mid - 1;
        temp[k++] = a[i++];
    }
    while (j <= r) temp[k++] = a[j++];


    for (int i = l, j = 0; i <= r; i++, j++) a[i] = temp[j];
}
int main()
{

    cin >> n;
    for (int i = 0; i < n; i++) scanf("%d", &a[i].x), a[i].y = i;
    merge(0, n - 1);

    LL res = 0;
    for (int i = 0; i < n; i++)  res += (1 + cnt[i]) * cnt[i] / 2;
    cout << res;
}

树状数组代码

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

typedef long long LL;

const int N = 1e6 + 10;

LL res;
int h[N], tr[N], 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 ans = 0;
    for (int i = x; i; i -= lowbit(i))  ans += tr[i];

    return ans;
}

int main() {
    int n;  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);
    }

    for (int i = 0; i < n; i ++)
        res += (LL)sum[i] * (sum[i] + 1) / 2;

    return cout << res << endl, 0;
}