周赛 厂里田径赛 题解(线段树)

题目链接

题目大意

其实就是找前面有多少个比他小的数

题目思路

显然要用线段树,是裸的线段树,但是不知道为什么自己却一点思路都没有。

对于能力值建立一颗线段树or 树状数组, 按编号从小到大枚举每个运动员,每次先查询当前能力值在[1,a[i]-1] 的人数

及第i个人的结果,然后再将这个人的能力值更新进去(让能力值为a[i]的人数+1).

注意询问的时候要加一句 if(L==R) return 0;要不然可能会死循环re

代码

#include<cstdio>
using namespace std;
const int maxn=3e5+5;
int n,a[maxn],tree[maxn<<2];
int query(int l,int r,int L,int R,int node)
{
    if(l<=L&&r>=R)
    {
        return tree[node];
    }
    if(L==R) return 0;//这一步判断不能忘掉,不然可能会死循环
    int ans=0,mid=(L+R)/2;
    if(mid>=l) ans+=query(l,r,L,mid,node<<1);
    if(mid<r)  ans+=query(l,r,mid+1,R,node<<1|1);
    return ans;
}
void update(int num,int l,int r,int node)//num表示要修改的值
{
    if(l==r)
    {
        tree[node]++;
        return ;
    }
    int mid=(l+r)/2;
    if(mid>=num) update(num,l,mid,node<<1);
    else         update(num,mid+1,r,node<<1|1);
    tree[node]=tree[node<<1]+tree[node<<1|1];
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        int ans=query(1,a[i]-1,1,3e5,1);
        printf("%d\n",ans);
        update(a[i],1,3e5,1);
    }
    return 0;
}

发布了68 篇原创文章 · 获赞 2 · 访问量 2278

猜你喜欢

转载自blog.csdn.net/m0_46209312/article/details/104770974