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

[bzoj3378][Usaco2004 Open]MooFest 狂欢节_树状数组

时间:2018-08-29 01:02:20      阅读:166      评论:0      收藏:0      [点我收藏+]

标签:highlight   class   add   algorithm   font   string   有一个   return   ios   

MooFest 狂欢节 bzoj-3378 Usaco-2004 Open

题目大意:给定一个n个数的a序列,每两个数之间有一个距离,两个点之间的权值为$max(a[i],a[j])*dis(i,j)$。

注释:$1\le n\le 2\cdot 10^4$。


想法:裙子说了,这种$max$和$min$的题通常要枚举这个$max$和$min$到底是多少。

这样的话我们就将所有点按权值从大到小排序。

往树状数组里插。

查询直接查询即可。

最后,附上丑陋的代码... ...

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm> 
using namespace std;
const int N=20000;
int n,num[N+5],cnt;
long long sum[N+5],ans,tot;
struct Node{int v,x;}a[N+5]; bool cmp(const Node &x,const Node &y){return x.v>y.v;}
inline int lowbit(int x) {return x&(-x);}
void add(int x,int v,int k)
{
    for(int i=x;i<=N;i+=lowbit(i))
        sum[i]+=v,num[i]+=k;
}
void query(int x)
{
    for(int i=x;i>0;i-=lowbit(i))
        tot+=sum[i],cnt+=num[i];
}
void work()
{
    for(int i=1;i<=n;i++)
	{
        tot=cnt=0;
        query(a[i].x);
        ans+=a[i].v*(cnt*a[i].x-tot);
        cnt=-cnt,tot=-tot;
        query(N);
        ans+=a[i].v*(tot-cnt*a[i].x);
        add(a[i].x,-a[i].x,-1);
    }
    cout << ans << endl ;
}
void init()
{
    cin >> n ;
    for(int i=1;i<=n;i++) scanf("%d%d",&a[i].v,&a[i].x);
    sort(a+1,a+1+n,cmp);
    for(int i=1;i<=n;i++) add(a[i].x,a[i].x,1);
}
int main()
{
    init();
    work();
    return 0;
}

小结:裙子有时候想法好$nb$啊!%%%

[bzoj3378][Usaco2004 Open]MooFest 狂欢节_树状数组

标签:highlight   class   add   algorithm   font   string   有一个   return   ios   

原文地址:https://www.cnblogs.com/ShuraK/p/9551668.html

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