hdu 4217Data Structure?

最新推荐文章于 2019-08-08 22:14:07 发布

asuxiexie

最新推荐文章于 2019-08-08 22:14:07 发布

阅读量709

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构搜索

本文链接：https://blog.youkuaiyun.com/asuxiexie/article/details/38088453

数据结构同时被 2 个专栏收录

30 篇文章

订阅专栏

搜索

16 篇文章

订阅专栏

本文介绍了一种结合树状数组与二分查找的技术，用于高效解决特定类型的问题，如寻找序列中第几小的元素并进行多次查询求和。通过C++实现，详细展示了算法的具体步骤和代码实现细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

树状数组+二分

就是找第几小的数，，找几次，再求和。。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
const int N=277777;
int t,n,m,bit[N],num,i;
long long ans;
int low(int g)
{
    return g&(-g);
}
void update(int pos)
{
    while(pos<=n)
    {
        bit[pos]--;
        pos+=low(pos);
    }
}
int query(int x)
{
    int sum=0;
    while(x>0)
    {
        sum+=bit[x];
        x-=low(x);
    }
    return sum;
}
int Sum(int k)
{
    int l=1,r=n,mid;
    while(l<r)
    {
        mid=(l+r)>>1;
        if(query(mid)>=k)
            r=mid;
        else
            l=mid+1;
    }
    return l;
}
void init()
{
    ans=0;
    memset(bit,0,sizeof(bit));
    scanf("%d %d",&n,&m);
    for(i=1;i<=n;i++)
        {bit[i]=low(i);}
    for(i=0;i<m;i++)
    {
        scanf("%d",&num);
        int s=Sum(num);
        update(s);
        ans+=s;
    }
}
int main()
{
    int cas=0;
    scanf("%d",&t);
    while(t--)
    {
        init();
        cout<<"Case "<<++cas<<": "<<ans<<endl;
    }
    return 0;
}

二分查找详解点击打开链接