Hoj 2275 Number Sequence

我发现传送门是个好东西:http://acm.hit.edu.cn/hoj/problem/view?id=2275

最近我正在做树状数组的问题,在做了数个裸题之后,我发现了这样一道题。

废话少说,上思路:

1.正序依次按如下步骤处理每个数据:用树状数组统计之前比它小的有多少,并记录在tmp数组里;从此数据开始向后更新更大的数。对此我要解释一下:树状数组的下标是数的大小,由于是按顺序来的,所以保证了i<j<k。由此可知道前面比此数小的有多少个。

2.将c数组清空后倒叙处理一遍数据。由此可知道后面比此数小的有多少个。

3.将每个数的两个数据相乘求和,最后用long long存储输出。

//以下是参照题解的结果 
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#define MAXN 50010
#define MAXM 32771 
int a[MAXN];
int c[MAXM];
int tmp[MAXN];
int n;
int lowbit(int x){
    return x&(-x);
}
void add(int x,int ad){
    while(x<MAXM){
	    c[x]+=ad;
        x+=lowbit(x);
    }
}
int sum(int x){
    int res=0;
    while(x>0){
        res+=c[x];
        x-=lowbit(x);
    } 
    return res;
}
int main(){
    while(scanf("%d",&n)!=EOF){ //这里坑我这么久你好意思么 
	    long long ans=0;
	    memset(tmp,0,sizeof(tmp));
	    memset(c,0,sizeof(c));
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
      	    a[i]++;
           	tmp[i]=sum(a[i]-1);   //注意是严格的大于
            add(a[i],1); 
        }
        memset(c,0,sizeof(c));
        for(int i=n;i>=1;i--){
        	ans+=(long long)sum(a[i]-1)*tmp[i]; //(long long)很按理说重要,不过没有加也过了。 
        	add(a[i],1);
        }
        printf("%lld\n",ans);
    }
    return 0;
}

值得一提的是 用while(scanf("%d",&n)会超时,一位朋友给的解释是它返回的值不一定是1,可若真如此,那有什么关系呢,真是匪夷所思。还望路过的神犇帮帮我这个蒟蒻。

猜你喜欢

转载自blog.csdn.net/u011431896/article/details/39033483