树状数组

这篇写的很详细
HDU1166

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
#include<cmath>
#define mem(a,b) memset(a,b,sizeof(a))
const int Max=5e4+10;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int mod=1e7+10;
int m,n;
int a[Max],c[Max];
int lowbit(int i)
{
    return i&(-i);
}
void update(int pos,int val)
{
    for(int i=pos; i<=n; i+=lowbit(i))
        c[i]+=val;
}
int getsum(int pos) //用前缀和 1-pos
{
    int ans=0;
    for(int i=pos; i>0; i-=lowbit(i))
        ans+=c[i];
    return ans;
}
int main()
{
    scanf("%d",&m);
    int k=0;
    while(m--)
    {
        scanf("%d",&n);
        mem(a,0);
        mem(c,0);
        printf("Case %d:\n",++k);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            update(i,a[i]);
        }
        char s[20]={0};
        int x,y;
        while(scanf("%s",s)&&s[0]!='E')
        {
            scanf("%d%d",&x,&y);
            if(s[0]=='Q')
            {
                printf("%d\n",getsum(y)-getsum(x-1));
            }
            else if(s[0]=='A')
                update(x,y);
            else if(s[0]=='S')
                update(x,-y);
        }
    }
    return 0;
}

树状数组求逆序对+离散化

#include<algorithm>
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int N = 5e5 + 10;
typedef long long ll;
#define lowbit(x) (x&(-x))
int n, m, c[N], a[N], b[N];
void add(int x)
{
    for (; x <= n; x += lowbit(x))
        c[x]++;
}
 int ask(int x)
{
    int s = 0;
    for (; x > 0; x -= lowbit(x))
        //cout << x <<" "<< c[x] << endl;
        s += c[x];
    return s;
}
bool cmp1(const int x, const int y)
{
    if (b[x] == b[y]) return x > y;//这个要注意,一定要让大的跑前面,不然就会多统计
    return b[x]>b[y];
}
int main()
{
    scanf("%d", &n);
    ll ans = 0;
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", &b[i]);
        a[i] = i;
    }
    sort(a + 1, a + n + 1, cmp1);
    for (int i = 1; i <= n; i++)
    {
        add(a[i]);
        ans += 1LL * ask(a[i] - 1);//之前加入数状数组的元素都比当前元素值大,因此我们查询下标比当前小的个数,累加答案
    }
    printf("%lld\n", ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值