前缀和(树状数组)构造桶数组的用法

咱们主要看两个题

  第一题数星星,大概意思就是一颗星星的左下方有几颗星星,这个星星就是几级,但是输入是特殊输入,输入是按y的升序,如果y一样按x的升序来进行输入,最会输出0-n-1级的星星各有几颗。

  因为后面的星星的高度一定大于等于前面的星星,所以就可以做一个实时的算法,也就是每输入一个星星就确定他的等级,因为之后的星星两种情况:

1.在它的上面,不算入等级。

2.与它同等高度,但x一定比它大,故也不算入等级。

  那么如何确定等级呢?其实我们现在的问题在于,当前这颗星星的高度一定大于等于前面的星星,但是x是不一定的,有可能高度大于,但x小于,所以我们关注点就不在高度,而是水平位置,我们的思路是构造一个桶数组,也就是下标代表星星的x坐标,然后数组的值代表数量,那么对于每个星星的操作,我们先不将它加入桶数组,我们(1)先求x坐标小于等于它的星星的前缀和,然后(2)再加入即可,那么我们简单分析一下时间复杂度,(1)为O(1),(2)为O(n),那么整体的时间复杂度就是O(n^2),显然是不行的,所以用树状数组优化为O(nlogn)即可。

天空中有一些星星,这些星星都在不同的位置,每个星星有个坐标。

如果一个星星的左下方(包含正左和正下)有 k 颗星星,就说这颗星星是 k 级的。

 

例如,上图中星星 5 是 33 级的(1,2,41,2,4 在它左下),星星 2,4 是 1 级的。

例图中有 1 个 0 级,2个 1 级,1 个 2 级,1 个 3 级的星星。

给定星星的位置,输出各级星星的数目。

换句话说,给定 N个点,定义每个点的等级是在该点左下方(含正左、正下)的点的数目,试统计每个等级有多少个点。

输入格式

第一行一个整数 N,表示星星的数目;

接下来 N行给出每颗星星的坐标,坐标用两个整数 x,y 表示;

不会有星星重叠。星星按 y 坐标增序给出,y 坐标相同的按 x坐标增序给出。

输出格式

N行,每行一个整数,分别是 0 级,1级,2 级,……N−1 级的星星的数目。

数据范围

1≤N≤150001≤N≤15000,
0≤x,y≤320000≤x,y≤32000

输入样例:

5
1 1
5 1
7 1
3 3
5 5

输出样例:

1
2
1
1
0
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
#define MAXN 32015
int level[MAXN],n,ans[MAXN];
int lowbit(int x)
{
    return x & -x;
}
void add(int x,int y)
{
    for(int i = x; i <= 32005; i += lowbit(i)) level[i] += y;
}
int query(int x)
{
    int res = 0;
    for(int i = x; i > 0; i -= lowbit(i)) res += level[i];
    return res;
}
int main()
{
    int x,y;
    scanf("%d",&n);
    for(int i = 1; i <= n; i++)
    {
        scanf("%d %d",&x,&y);
        add(x+1,1);
        ans[query(x+1)]++;
    }
    for(int i = 1; i <= n; i++) printf("%d\n",ans[i]);
    return 0;
}

 第二题,小朋友排队,原理跟数星星是一样的,但是得分析一下。题目大概的意思就是一个乱序的序列,问我们恢复到升序的总次数,并且每一项每被移动一次,代价会加一,也就是我们需要弄明白每一项到底被移动了几次,那么其实这里有一个小知识,就是逆序对的问题,逆序对与乱序恢复到升序的最少移动次数是等价的,例如 3 2 1中的逆序对有 (3,2)(3,1)(2,1) 移动的时候,3->2,3->1,2->1三次,那么再细致一点,我们研究某一项在这个过程中被移动的次数与逆序对的关系,5 4 3 2 1,对于三的逆序对有(5,3) (4,3) (3,2) (3,1) 所以3会被移动4次,其实你可以这样理解,3在中间,1,2要归位必须穿过3往左走,5,4要归位必须穿过3往右走,所以我们只需要找3的左边比3大的数,3的右边比3小的数,那么其实我们又回到了线上树状数组优化的桶数组,也就是每进来一个数,我们就看看前面比他大的前缀和有多少个,即query(MAXN - 1) - query(h[i]),那么逆序的时候,我们就看看前面有没有比他下的即可,即query(h[i]-1)。

n 个小朋友站成一排。

现在要把他们按身高从低到高的顺序排列,但是每次只能交换位置相邻的两个小朋友。

每个小朋友都有一个不高兴的程度。

开始的时候,所有小朋友的不高兴程度都是 00。

如果某个小朋友第一次被要求交换,则他的不高兴程度增加 11,如果第二次要求他交换,则他的不高兴程度增加 22(即不高兴程度为 33),依次类推。当要求某个小朋友第 kk 次交换时,他的不高兴程度增加 kk。

请问,要让所有小朋友按从低到高排队,他们的不高兴程度之和最小是多少。

如果有两个小朋友身高一样,则他们谁站在谁前面是没有关系的。

输入格式

输入的第一行包含一个整数 n,表示小朋友的个数。

第二行包含 nn 个整数 H1,H2,…,Hn分别表示每个小朋友的身高。

输出格式

输出一行,包含一个整数,表示小朋友的不高兴程度和的最小值。

数据范围

1≤n≤100000
0≤Hi≤1000000

输入样例:

3
3 2 1

输出样例:

9

样例解释

首先交换身高为3和2的小朋友,再交换身高为3和1的小朋友,再交换身高为2和1的小朋友,每个小朋友的不高兴程度都是3,总和为9。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define MAXN 1000015
#define N 100015
typedef long long ll;
int n,tr[MAXN],h[N],sum[N];
int lowbit(int x)
{
    return x & -x;
}
void add(int x,int c)
{
    for(int i = x; i < MAXN; i += lowbit(i)) tr[i] += c;
}
int query(int x)
{
    int res = 0;
    for(int i = x; i; i -= lowbit(i)) res += tr[i];
    return res;
}
int main()
{
    scanf("%d",&n);
    for(int i = 1; i <= n; i++) scanf("%d",&h[i]);
    for(int i = 1; i <= n; i++)
    {
        sum[i] += query(MAXN-1) - query(h[i]);
        add(h[i],1);
    }
    memset(tr,0,sizeof(tr));
    for(int i = n; i > 0; i--)
    {
        sum[i] += query(h[i]-1);
        add(h[i],1);
    }
    ll res = 0;
    for(int i = 1; i <= n; i++) res += (ll)sum[i] * (1 + sum[i]) / 2;
    cout << res << endl;
    return 0;
}

 

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值