小奇遐想树状数组实现+容斥思想

最新推荐文章于 2019-04-30 21:24:06 发布

原创最新推荐文章于 2019-04-30 21:24:06 发布 · 260 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

ACM

本文介绍了一个有趣的算法问题：如何计算一组高度不同的白云中能够形成彩虹形状的组合数量。该问题涉及容斥原理及树状数组的应用，通过巧妙的数学转换简化了问题复杂度。

问题 M: 小奇遐想

时间限制: 1 Sec 内存限制: 128 MB
提交: 165 解决: 21
[提交] [状态] [讨论版] [命题人:admin]

题目描述

撷来一缕清风飘渺
方知今日书信未到
窗外三月天霁垂柳新长枝条
风中鸟啼犹带欢笑
——《清风醉梦》
小奇望着青天中的悠悠白云，开始了无限的遐想，在它的视野中，恰好有n朵高度不同的白云排成一排，他想从左到右选出四朵白云a，b，c，d，使得h_a<h_b<h_d<h_c，即看起来像是彩虹的形状！它想知道有多少种方案数。

输入

第一行包括1个整数n。
第二行包括n个整数，第i个正数表示h_i，保证这n个整数是n的一个全排列。

输出

输出一个整数表示答案。（mod 16777216）

样例输入

5
1 5 3 2 4

样例输出

提示

对于10%的数据n<=600；对于40%的数据n<=5000；
对于100%的数据n<=200000。

在一序列中求满足1243大小关系的串

利用容斥的思想，转换为先求12xx（包括1234，1243），然后再减去1234

先利用一次树状数组求出每个数前面比他小的数字，存在l[i]数组里

通过计算得到每个数后面比他大的数字，存在r[i]数组里

12xx很好计算，就是取数相乘即可

然后计算1234，因为每个数字比他小的已经存在l[i]数组里，所以一种想法是用树状数组求一次比他大的，然后在求一次

优化之后的是利用l[i]数组计算每个数前面的不下降子串的个数，然后再乘后面比他大的数字个数

最后两者相减

（好不容易有正确的想法，结果因为不会用树状数组求1234，卡死在这道题上）

最后记得取余，还有不是取1e9+7

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxx=2e5+100;
const int INF=1e9;
const int MOD=16777216;
int n;
int a[maxx];
int l[maxx],r[maxx];
int sum[maxx];
int ask(int x)
{
    int ret=0;
    for(int i=x; i>0; i-= i&(-i)){
        ret+=sum[i];
    }
    return ret;
}
void add(int x,int c)
{
    for(int i=x; i<=n; i+= i&(-i)){
        sum[i]+=c;
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1; i<=n; i++){
        cin>>a[i];
        l[i]=ask(a[i]);
        r[i]=n-i-a[i]+1+l[i];
        add(a[i],1);
    }
    ll ans1=0;
    for(int i=1; i<=n; i++){
        ans1=(ans1+l[i]*r[i]*(r[i]-1)/2)%MOD;
    }
    memset(sum,0,sizeof(sum));
    ll ans2=0;
    for(int i=1; i<=n; i++){
        ans2=(ans2+ask(a[i])*r[i])%MOD;
        add(a[i],l[i]);
    }
    printf("%lld\n",(ans1-ans2+MOD)%MOD);
    return 0;
}