清点人数-树状数组

最新推荐文章于 2023-03-23 17:15:30 发布

原创最新推荐文章于 2023-03-23 17:15:30 发布 · 1.5k 阅读

1 ·

CC 4.0 BY-SA版权

c/c++ 同时被 3 个专栏收录

83 篇文章

订阅专栏

算法

21 篇文章

订阅专栏

树状数组

4 篇文章

订阅专栏

这题我用了两种方法，都超时，最后百度一个出来.....，上传两个代码，第一个代码是网上查的，那三个函数表示有点不懂，求解答！

描述

初始时，火车上没有学生；当同学们开始上火车时，年级主任从第一节车厢出发走到最后一节车厢，每节车厢随时都有可能有同学上下。年级主任走到第m节车厢时，他想知道第１到m这m节车厢上一共有多少学生，但是他没有调头往回走的习惯．也就是说每次当他提问时，m总会比前一次大。

输入

第一行两个整数n,k，表示火车共有n节车厢以及k个事件。接下来有k行，按时间先后给出k个事件，每行开头都有一个字母A,B或C，如果字母为A，接下来是一个数m，表示年级主任现在在第m节车厢；如果为B，接下来两个数m,p，表示在第m节车厢有p名学生上车；如果为C，接下来两个数m,p，表示在第m节车厢有p名学生下车。学生总人数不会超过100000。

输出

有多少个A就输出多少行，每行一个整数，回答年级主任提出的问题。

样例输入

10 7
A 1
B 1 1
B 3 1
B 4 1
A 2
A 3
A 10

样例输出

0
1
2
3

提示

注意:对于30%的数据,n< =10000,k< =10000 至少有3000个A 对于100%的数据n< =500000,k< =100000. 至少有30000个A

#include<iostream>
#include<string.h>
#include<stdio.h>
#include<stdlib.h>
long c[500001];
long n,k;
long fun(long x)
{
    return x^(x&(x-1));
}
void modify(long x,long d)
{
    while (x<=n)
    {
        c[x]+=d;
        x+=fun(x);
    }
}
long getsum(long x)
{
    long s=0;
    while (x)
    {
        s+=c[x];
        x-=fun(x);
    }
    return s;
}
int main()
{
    scanf("%d%d",&n,&k);
    getchar();
    memset(c,0,sizeof(c));
    long x,y;
    char c;
    for (int i=1;i<=k;i++)
    {
        scanf("%c",&c);

        if (c=='A')
        {
            scanf("%d",&x);
            getchar();
            printf("%d\n",getsum(x));
        }
        else if (c=='B')
        {
            scanf("%d%d",&x,&y);
            getchar();
            modify(x,y);
        }
        else
        {
            scanf("%d%d",&x,&y);
            getchar();
            modify(x,-y);
        }
    }
    return 0;
}


/*#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int a[1000000];
int main()
{
    int n,k;
    char c;
    while(scanf("%d %d",&n,&k)!=EOF)
    {
        int sum=0;
        memset(a,0,sizeof(a));
        for(int i=0;i<k;i++)
        {

            cin>>c;
            if(c=='A')
            {
                int ms;
                scanf("%d",&ms);
                printf("%d\n",a[ms]);
            }
            if(c=='B')
            {
                int ms,p;
                scanf("%d%d",&ms,&p);
                for(int ii=ms;ii<=n;ii++)
                   a[ii]+=p;
            }
            if(c=='C')
            {
                int ms,p;
                scanf("%d%d",&ms,&p);
                for(int ii=ms;ii<=n;ii++)
                   a[ii]-=p;
            }
        }
    }
    return 0;
}
*/