设为首页 加入收藏

TOP

HIT 2275 Number sequence
2014-11-23 19:30:28 来源: 作者: 【 】 浏览:12
Tags:HIT 2275 Number sequence

思路: 树状数组

分析:

1 题目要求的是总共的搭配方式,满足Ai < Aj > Ak.并且i j k不同

2 我们开两个树状数组,第一个在输入的时候就去更新。然后我们在去枚举Aj 同时维护第二个树状数组,对于AI来说就是在第二个树状数组里面求和

然后在通过第一个树状数组就可以求出Ak的个数,把结果相乘即可


代码:

#include
#include
#include
#include
using namespace std;

const int MAXN = 50010;

int n , num[MAXN];
int treeNumOne[MAXN];
int treeNumTwo[MAXN];

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

int getSum(int *arr , int x){
    int sum = 0;
    while(x){
        sum += arr[x];
        x -= lowbit(x);
    }
    return sum;
}

void add(int *arr , int x , int val){
    while(x < MAXN){
         arr[x]  += val; 
         x += lowbit(x);
    }
}

long long getAns(){
    if(n < 3)
        return 0;
    long long ans = 0;
    add(treeNumTwo , num[1] , 1);
    for(int i = 2 ; i < n ; i++){
        int x = getSum(treeNumTwo , num[i]-1); 
        int y = getSum(treeNumOne , num[i]-1); 
        add(treeNumTwo , num[i] , 1);
        ans += (x)*(y-x); 
    }
    return ans;
}

int main(){
    while(scanf("%d" , &n) != EOF){
         memset(treeNumOne , 0 , sizeof(treeNumOne));
         memset(treeNumTwo , 0 , sizeof(treeNumTwo)); 
         for(int i = 1 ; i <= n ; i++){
             scanf("%d" , &num[i]);  
             num[i]++;
             add(treeNumOne , num[i] , 1);
         }
         printf("%lld\n" , getAns());
    }
    return 0;
}

】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
分享到: 
上一篇POJ 3984 迷宫问题 (DFS(深度优先.. 下一篇uva 644

评论

帐  号: 密码: (新用户注册)
验 证 码:
表  情:
内  容:

·求navicat for mysql (2025-12-26 13:21:33)
·有哪位大哥推荐一下m (2025-12-26 13:21:30)
·MySQL下载与安装教程 (2025-12-26 13:21:26)
·Linux_百度百科 (2025-12-26 12:51:52)
·Shell 流程控制 | 菜 (2025-12-26 12:51:49)