码迷,mamicode.com
首页 > 编程语言 > 详细

HDU- 4000 Fruit Ninja(树状数组)

时间:2017-05-31 12:04:10      阅读:159      评论:0      收藏:0      [点我收藏+]

标签:开始   log   bsp   eof   const   通过   sum   type   sizeof   

题目大意


给出一串数字,是1到n的一种排列,找出  x<z<y 的三个数字的个数(在数列中x,y,z按顺序出现)

分析


刚刚拿到题目,就有一种树状数组处理比当前数字大或者小的感觉,一开始我锁定了y,在y的右边和左边分别找到比y小的数字,然而这样只能保证 z<y,x<y,不能保证x<z。所以最终获得的三个数字会有x<z<y,与z<x<y两种可能。筛去z<x<y并不好处理。

所以:

如果锁定x寻找右边比x大的数字,可以得到x<y<z,以及x<z<y两种可能。x<y<z可通过固定y寻找左边比它小,右边比它大来计数的方法直接筛去。

注意乘法时候的数据范围。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define MAX_NUM            100005
#define modd             100000007
#define lowbit(q)            (q)&(-(q))
typedef long long int LL;
int c[MAX_NUM];
void update(int i ){
    for (int j = i ; j <= 1e5 ; ){
        c[j] ++ ;
        j += lowbit(j);
    }
}
int query(int i){
    int sum = 0; 
    while(i){
        sum += c[i];
        i -= lowbit (i);
    }
    return sum ;
}
LL num[MAX_NUM];
LL pre[MAX_NUM];
LL las[MAX_NUM];
int main(int argc, char const *argv[])
{
    int t;
    scanf("%d",&t);
    int vs = 0;
    while(t--){
        int n;
        scanf("%d",&n);
        LL ans = 0;
        memset(c,0,sizeof(c));
        for(int i = 1 ; i <= n; i++){
            scanf("%lld",&num[i]);
            pre[i] = query(num[i]-1);
            update(num[i]);
        }
        memset(c,0,sizeof(c));
        for(int i = n ; i >0 ; i--){
            las[i] = (n-i) - query(num[i]);
            ans =(ans + (las[i]*(las[i]-1))/2 - (pre[i]*las[i]) )%modd;
            update(num[i]);
        }
        printf("Case #%d: %lld\n",++vs,ans );
    }
    return 0;
}

 

HDU- 4000 Fruit Ninja(树状数组)

标签:开始   log   bsp   eof   const   通过   sum   type   sizeof   

原文地址:http://www.cnblogs.com/miamiao/p/6922823.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!