【python】第七天打卡 离散化

于妍 2023-02-07 23:21:21

 思路

代码

def find(x):
    """二分查找模板,从索引数组alls中找到大于等于x的最小的索引"""
    l = 0
    r = len(alls)-1
    while l<r:
        mid = l+r>>1
        if alls[mid]>=x: r = mid    # !!!if条件忘记了=号
        else: l = mid+1
    return l+1    # 因为要计算前缀和,所以加1保证索引从1开始

if __name__=="__main__":
    n, m = map(int, input().split())
    N = 300010
    a = [0]*N    # 用于存储离散化后的索引和对应值,其中索引对应离散化后的索引,值对应离散化前索引的取值
    s = [0]*N    # 存a数组的前缀和数组

    add = []    # 存储插入操作的二元组
    query = []    # 存储查询操作的二元组

    alls = []    # 存储离散化前输入的所有索引,n+2*m

    for i in range(n):
        x, c = map(int, input().split())
        add.append((x, c))
        alls.append(x)

    for i in range(m):
        l, r = map(int, input().split())
        query.append((l, r))
        alls.append(l)
        alls.append(r)

    alls = list(set(sorted(alls)))    # 将alls数组排序并去重


    # 1. 处理插入
    for x, c in add:
        x2 = find(x)
        a[x2]+=c

    # 2. 处理前缀和
    for i in range(1, len(alls)+1):
        s[i] = s[i-1]+a[i]

    # 3. 处理查询
    for l, r in query:
        l2 = find(l)
        r2 = find(r)
        res = s[r2]-s[l2-1]
        print(res)

 总结

 

...全文
8 回复 打赏 收藏 转发到动态 举报
写回复
用AI写文章
回复
切换为时间正序
请发表友善的回复…
发表回复

30,346

社区成员

发帖
与我相关
我的任务
社区描述
打造最热爱学习的高校社区,帮助大家提升计算机领域知识,带大家打比赛拿奖,提高自我,希望大家共同创造良好的社区氛围。
社区管理员
  • 川川菜鸟
  • 亡心灵
  • 星辰菜鸟
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告

监督大家学习,每日学习打卡,以投稿形式打卡。扫码关注公众号,可加入粉丝群和领取大量资源。

试试用AI创作助手写篇文章吧