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

POJ2299题解(树状数组)

时间:2019-12-28 23:08:37      阅读:92      评论:0      收藏:0      [点我收藏+]

标签:upd   应用   文章   次数   struct   表示   org   node   ret   

POJ2299题解

2019-12-28

Powered by Gauss

1.题目传送门:POJ2299

 

2.理解题意:

做题的时候,第一步也是最重要的就是理解题意。记住这句话。

POJ2299是一道标准的树状数组模板题。题意大致如下:

快速排序是一种非常优秀的排序方式,其精华在于循环比较和swap函数的应用。

先给你一个数N,表示这组数据(没错,这题是多组数据)数的个数,之后的N行,每行一个数Ai

请你输出在快速排序过程中经历的swap函数的次数。

3.算法思路:

树状数组是一种非常高效的数据结构,由于篇幅有限,这里不做讲解,作者将尽快写出一篇树状数组的稿子,敬请谅解。

树状数组的精华在于lowbit函数的使用,为了迎合题目,这里将query函数和update函数做了一些改动。

void update(int x,int k)
{
    int i;
    for(i=x;i<=n;i=i+lowbit(i))
    {
        tree[i]+=k;
    }
}
int query(int x)
{
    int i,sum=0;
    for(i=x;i>0;i=i-lowbit(i))
    {
        sum+=tree[i];
    }
    return sum;
}

我们因为这道题的数据规模非常大,所以要用到一些离散化的思想。

for(i=1;i<=n;i++)
{
     cin>>a[i].value;
     a[i].id=i;
}
sort(a+1,a+n+1,cmp);
for(i=1;i<=n;i++)
{
    a[i].pos=i;
}

有了离散化,我们就可以大大降低时间复杂度了。

上面那段代码中,我们用结构体node来实现,包含三个参数,id,pos和value。

struct node
{
  int id,value,pos;
}a[500001];

最后,结合树状数组的计算模板,我们给出最后的AC源代码,用时3610MS:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int tree[500001],n;
struct node
{
  int id,value,pos;
}a[500001];
int lowbit(int x)
{
    return (x&(-x));
}
void update(int x,int k)
{
    int i;
    for(i=x;i<=n;i=i+lowbit(i))
    {
        tree[i]+=k;
    }
}
int query(int x)
{
    int i,sum=0;
    for(i=x;i>0;i=i-lowbit(i))
    {
        sum+=tree[i];
    }
  return sum;
}
int cmp(node a,node b)
{
    return a.value<b.value;
}
int cmp1(node a,node b)
{
    return a.id<b.id;
}
int main()
{
    int i;
    long long sum;
    while(cin>>n && n)
    {
        sum=0;
        memset(tree,0,sizeof(tree));
        for(i=1;i<=n;i++)
        {
            cin>>a[i].value;
            a[i].id=i;
        }
        sort(a+1,a+n+1,cmp);
        for(i=1;i<=n;i++)
        {
            a[i].pos=i;
        }
        sort(a+1,a+n+1,cmp1);
        for(i=1;i<=n;i++)
        {
            sum+=i-1-query(a[i].pos);
            update(a[i].pos,1);
        }
        cout<<sum<<endl;
    }
    return 0;
}

版权声明:知识产权神圣不可侵犯,这篇文章中的部分代码和思想源自:https://blog.csdn.net/weixin_43918531/article/details/87950037

POJ2299题解(树状数组)

标签:upd   应用   文章   次数   struct   表示   org   node   ret   

原文地址:https://www.cnblogs.com/Warframe-Gauss/p/12113028.html

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