HDU4000Fruit Ninja【树状数组+组合数】

大意:

告诉你一个有n个数的序列 (1 -- n) 问其中有多少组 (a[i], a[j], a[k]) 满足i < j < k 并且 a[i] < a[k] < a[j]

 

分析:

这个题跟那个中间为峰值的题很像   我的第一反应为树状数组

求的就是在这个序列当中  小大中 一共会出现多少次

小大中 = (小中大 + 小大中) — 小中大

(小中大 + 小大中) 也就是说只要前面这个数是小的  后面的大的数中任取两个就可以了

用两个数组来维护

一个qian[]用来存前边有多少个小于这个数  另一个hou[]维护后面有多少个大于这个数

小中大的个数就是  sum(qian[i] * hou[i])==

求小XX的过程中我脑袋抽筋了

我一开始用的是这种思路

对于每一位

用比它小的个数 * (C(大于它的g个数, 2))  

看起来貌似很合理

但是其实有个严重的问题

在求小中大的过程我们可以那么考虑是因为中间的是唯一的

所以无论我们怎么成都不会重复

但是这个若只是考虑小于它的和大于等于它的想成就会出现重复的

怎么避免重复呢

只要按照求小中大的思路

我们可以确定一个最小的  那么  之后的就是以这个为基准的 而不会出现重复

小XX = sum(C(大于该位, 2) );

 

代码:

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 using namespace std;
 5 
 6 const long long maxn = 100005;
 7 const long long mod = 100000007;
 8 long long c[maxn];
 9 long long n, a[maxn];
10 long long qian[maxn], hou[maxn], hh[maxn];
11 
12 long long lowbit(long long x) {
13     return x & ( - x ) ;
14 }
15 
16 void add(long long i) {
17     while(i <= n) {
18         c[i] += 1;
19         i += lowbit(i);
20     }
21 }
22 
23 long long sum(long long i) {
24     long long s = 0;
25     while(i > 0) {
26         s += c[i];
27         i -= lowbit(i);
28     }
29     return s;
30 }
31 
32 void init() {
33     memset(c, 0, sizeof(c));
34     for(long long i = 1; i <= n; i++) {
35         qian[i] = sum(a[i]);
36         add(a[i]);
37     }
38     memset(c, 0, sizeof(c));
39     for(long long i = n; i >= 1; i--) {
40         add(a[i]);
41         hou[i] = n - i + 1 - sum(a[i]);
42     }
43 }
44 
45 int main() {
46     long long t;
47     scanf("%I64d",&t);
48     for(long long kase = 1; kase <= t; kase++) {
49         scanf("%I64d",&n);
50         for(long long i = 1; i <= n; i++) scanf("%I64d",&a[i]);
51         init();
52         long long sum1 = 0;
53         for(long long i = 1; i <= n; i++) {
54             sum1 += qian[i] * hou[i];
55             sum1 %= mod;
56         }
57         long long sum2 = 0;
58         for(long long i = 1; i <= n; i++) {
59             sum2 += (hou[i] * (hou[i] - 1) / 2);
60             sum2 %= mod;
61         }
62         printf("Case #%I64d: ", kase);
63         long long ans = (sum2 - sum1 + mod) % mod;
64         printf("%I64d\n", ans);
65     }
66     return 0;
67 }
View Code

 

posted @ 2014-10-17 19:44  悠悠我心。  阅读(148)  评论(0编辑  收藏  举报