关闭

POJ 2299 树状数组

67人阅读 评论(0) 收藏 举报
分类:

第一次自己在题里遇到了离散化,也就是这个东西困了我好久的,离散化一般的处理感觉好多种的写法,但是有好有坏,选择合适的写法写即可。

1.

        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&a[i]);
            p.push_back(a[i]);
        }
       sort(p.begin(),p.end());
        p.erase(unique(p.begin(),p.end()),p.end());//如果要删除相同的元素的话
        for(int i=0;i<p.size();i++)
        {
            mp[p[i]]=i+1;
        }

2.

struct Node
{
    int pos,v;
}num[maxn];

bool cmp(Node a,Node b)
{
    return a.v<b.v;
}

  for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
         num[i].v=a[i];
         num[i].pos=i;
       }
        sort(num+1,num+1+n,cmp);
        for(int i=1;i<=n;i++)
        {
          x[num[i].pos]=i;
        }

个人感觉第一种写法更加好理解,第二种同样是可以的,第一种利用的是map来直接对应元素a[i]和离散数值i,然后第二种通常通过结构体来设立i然后由得原来顺序的pos来对应好元素i。

在这个题里面感觉第一种那么写,不知道为什么就是一直超时。(搞不明白= =)、、望大神路过解答下,不胜感激。。

题目的思路还是简单的, 在树状数组上面数列的第i个数了,比i小的包括i的数据有几个为sum也就是c[i],然后相减为前面的比i大的个数。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<vector>

using namespace std;

const int maxn=500010;
int n,a[maxn],c[maxn],x[maxn];
struct Node
{
    int pos,v;
}num[maxn];

int lowbit(int x)
{
    return x&(-x);
}

int sum(int x)
{
  int ret=0;
  while(x>0)
  {
      ret+=c[x];
      x-=lowbit(x);
  }
  return ret;
}

void adds(int x,int d)
{
    while(x<=n)
    {
        c[x]+=d;
        x+=lowbit(x);
    }
}

bool cmp(Node a,Node b)
{
    return a.v<b.v;
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        if(!n)
            break;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            num[i].v=a[i];
            num[i].pos=i;

        }
        sort(num+1,num+1+n,cmp);
        for(int i=1;i<=n;i++)
        {
          x[num[i].pos]=i;
        }
        memset(c,0,sizeof(c));
        long long ans=0;
        for(int i=1;i<=n;i++)
        {
            adds(x[i],1);
            ans+=i-sum(x[i]);//数列的第i个数了,比i小的包括i的数据有几个为sum,然后相减为前面的比i大的个数
        }
         printf("%lld\n",ans);

    }
    return 0;
}

还有一种归并排序的写法,也先沾上



#include <iostream>  
#include <cstdio>  
using namespace std;  
  
long long num[500005];  
long long temp[500005];  
int n;  
long long ans;  
  
void merge(int low, int mid, int high)  
{  
    int i = low, j = mid + 1, k = low;  
    while (i <= mid && j <= high)  
    {  
        if (num[i] <= num[j])   
        {  
            temp[k++] = num[i++];  
        }  
        else   
        {  
            ans += j - k ;<span style="white-space:pre">    </span>//统计逆序  
            temp[k++] = num[j++];  
        }  
    }  
    while (i <= mid) temp[k++] = num[i++];  
    while (j <= high) temp[k++] = num[j++];  
    for (i = low; i <= high; ++i)  
    {  
        num[i] = temp[i];  
    }  
}  
  
void mergeSort(int a, int b)  
{  
    if (a < b)  
    {  
        int mid = (a + b) / 2;  
        mergeSort(a, mid);  
        mergeSort(mid + 1, b);  
        merge(a, mid, b);  
    }  
}  
  
  
int main()  
{  
    while (scanf("%d", &n) != EOF && n > 0)  
    {  
        ans = 0;  
        for (int i = 0; i < n; ++i)  
        {  
            scanf("%lld", num + i);  
        }  
        mergeSort(0, n - 1);  
        printf("%lld\n", ans);  
    }  
    return 0;  
}  
来自http://blog.csdn.net/alongela/article/details/8119209点击打开链接



0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:8450次
    • 积分:1053
    • 等级:
    • 排名:千里之外
    • 原创:97篇
    • 转载:25篇
    • 译文:0篇
    • 评论:4条
    最新评论
  • HDU 1536 S-Nim

    qq_23702679: http://www.cnblogs.com/kuangbin/archive/2011/08/28...

  • HDU 1114

    qq_23702679: 还有就是这个是完全背包里求最小价值的,有那么一点点不同于背包九讲里的做法,这个还是可以好好想下的,首...

  • HDU 1114

    qq_23702679: 还有就是这个是完全背包里求最小价值的,有那么一点点不同于背包九讲里的做法,这个还是可以好好想下的,首...

  • HDU 2612

    qq_23702679: 特别最后忘记了判断了YM都是可以共同达到改点@这个要求,造成了我很多的错误。