您的位置:首页 > 产品设计 > UI/UE

HOJ 2275 Number sequence(树状数组)

2015-02-23 10:45 351 查看


题意:给定一个n个元素的数列,令Ai, Aj, Ak,使得 Ai < Aj > Ak 且 i < j < k这样的三个数为一组。求出一共有多少组。

思路:可以用树状数组,每次输入一个Ai,可以查询到之前输入的比它小或比它大的有多少组,之后输入的就不得而知了,所以可以开个数组记录下来逆序再建树一次即可
另外数的范围取到了0,所以每个数要自加一次,而且组数是longlong的,而且HOJ的服务器系统支持的是lld,不是i64d..wa了几次
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
#define M 35000
#define N 50050
#define ll long long
#define lowbit(x) (x&-x)
int tree[M];
int num
;
ll tmp
;
int sum(int rt)
{
    int s=0;
    while(rt>0){
        s+=tree[rt];
        rt-=lowbit(rt);
    }
    return s;
}
void update(int rt)
{
    while(rt<=M-1){
        tree[rt]++;
        rt+=lowbit(rt);
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n)){
        memset(tree,0,sizeof(tree));
        for(int i=1;i<=n;i++){
            scanf("%d",&num[i]);
            num[i]++;
        }
        ll ans=0;
        for(int i=1;i<=n;i++){
            tmp[i]= sum(num[i]-1);
            update(num[i]);
        }
        memset(tree,0,sizeof(tree));
        for(int i=n;i>=1;i--)
        {
            ans+=tmp[i]*(ll)sum(num[i]-1);
            update(num[i]);
        }
        printf("%lld\n",ans);
    }
    return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: