CDOJ 数据结构训练 E

最新推荐文章于 2017-08-20 00:34:16 发布

原创最新推荐文章于 2017-08-20 00:34:16 发布 · 444 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树状数组 #数据结构

CDOJ 专栏收录该内容

32 篇文章

订阅专栏

本文介绍了一个基于C++实现的区间树并查集算法，通过使用杂凑表来快速查找和更新区间内元素个数，进而计算特定条件下两数之间的组合数量。文章详细展示了算法的实现过程，并提供了完整的代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <map>
#include <set>
#include <list>
#include <cmath>
#include<cctype>
#include <ctime>
#include <deque>
#include <stack>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long
#define PI 3.1415926535897932626
#define MAXN 50005
#define MAXD 32800
using namespace std;
int gcd(int a, int b)
{
	return a % b == 0 ? b : gcd(b, a % b);
}
int C[MAXD];
LL ans;
int N;
int Left[MAXN],num[MAXN];
inline void insert(int pos)//注意这里是利用杂凑以下标与对应值判断元素个数，方便计算最后结果
{
    while (pos<=MAXN)
    {
        C[pos]++;
        pos+=(pos&(-pos));
    }
}
inline int sum(int x)//这里并不是直接求元素的和，而是利用下标对应关系求的是元素个数的和
{
    int sum=0;
    while (x>0)
    {
        sum+=C[x];
        x-=(x&(-x));
    }
    return sum;
}
int main()
{
    memset(C,0,sizeof(C));
    scanf("%d",&N);
    for (int i=1;i<=N;i++)
    {
        scanf("%d",&num[i]);
        Left[i]=sum(num[i]-1);//比当前数字小的存入left【i】，下标从小到大
        insert(num[i]);//当前数字对应的区间全部加1
    }
    ans=0;
    //逆向
    memset(C,0,sizeof(C));
    for (int i=N;i>=1;i--)//下面从大至小
    {
        ans=ans+(Left[i]*sum(num[i]-1));
        insert(num[i]);
    }
    printf("%lld\n",ans);
    return 0;
}