poj 1990 树状数组

开始对距离排序 然后果断的TLE了

然后发现应该对power排序,然后我们记录和更新的是 比x小的个数与距离和,然后每次算的时候,因为按power排序了,当前的pi一定是最大的,所以就能在logdistance的时间内算出来

AC代码如下:

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;

const int MAX_X = 20000 + 5;

struct Node{
    long long v, d;
};

long long sum[MAX_X], num[MAX_X];
int N;
Node node[MAX_X];

bool cmp( Node a, Node b ){
    if( a.v < b.v ){
        return true;
    }else{
        return false;
    }
}

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

long long getsum( int x ){
    int ans = 0;
    while( x > 0 ){
        ans += sum[x];
        x -= lowbit( x );
    }
    return ans;
}

long long getnum( int x ){
    int ans = 0;
    while( x > 0 ){
        ans += num[x];
        x -= lowbit( x );
    }
    return ans;
}

void updata( int x, long long d ){
    while( x <= MAX_X ){
        sum[x] += d;
        num[x] += 1;
        x += lowbit( x );
    }
}

int main(){
    while( scanf( "%d", &N ) != EOF ){
        for( int i = 1; i <= N; i++ ){
            scanf( "%lld%lld", &node[i].v, &node[i].d );
        }
        sort( node + 1, node + 1 + N, cmp );
        memset( sum, 0, sizeof( sum ) );
        memset( num, 0, sizeof( num ) );
        int t[MAX_X];
        t[0] = 0;
        for( int i = 1; i <= N; i++ ){
            t[i] = t[i-1] + node[i].d;
        }
        long long ans = 0;
        for( int i = 1; i <= N; i++ ){
            int t1 = getsum( node[i].d );
            int t2 = getnum( node[i].d );
            int t3 = t[i-1] - t1;
            int t4 = i - 1 - t2;
            ans += node[i].v * ( t2 * node[i].d - t1 + t3 - t4 * node[i].d );
            updata( node[i].d, node[i].d );
        }
        printf( "%lld\n", ans );
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值