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

Posted 2021-01-02 shurak

MooFest 狂欢节 bzoj-3378 Usaco-2004 Open

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

注释：$1le nle 2cdot 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$啊！%%%