2 Star 15 Fork 5

低调之壮志凌云/fucking-algorithm

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
.github/ISSUE_TEMPLATE
pictures
动态规划系列
技术
数据结构系列
算法思维系列
FloodFill算法详解及应用.md
README.md
UnionFind算法应用.md
UnionFind算法详解.md
twoSum问题的核心思想.md
为什么推荐算法4.md
二分查找详解.md
信封嵌套问题.md
几个反直觉的概率问题.md
前缀和技巧.md
区间交集问题.md
区间调度问题之区间合并.md
双指针技巧.md
回溯算法详解修订版.md
字符串乘法.md
学习数据结构和算法的高效方法.md
常用的位操作.md
洗牌算法.md
滑动窗口技巧.md
烧饼排序.md
算法学习之路.md
递归详解.md
高频面试系列
README.md
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
区间交集问题.md 3.65 KB
一键复制 编辑 原始数据 按行查看 历史
labuladong 提交于 5年前 . 添加页脚跳转

区间交集问题

本文是区间系列问题的第三篇,前两篇分别讲了区间的最大不相交子集和重叠区间的合并,今天再写一个算法,可以快速找出两组区间的交集。

先看下题目,LeetCode 第 986 题就是这个问题:

title

题目很好理解,就是让你找交集,注意区间都是闭区间。

思路

解决区间问题的思路一般是先排序,以便操作,不过题目说已经排好序了,那么可以用两个索引指针在 AB 中游走,把交集找出来,代码大概是这样的:

# A, B 形如 [[0,2],[5,10]...]
def intervalIntersection(A, B):
    i, j = 0, 0
    res = []
    while i < len(A) and j < len(B):
        # ...
        j += 1
        i += 1
    return res

不难,我们先老老实实分析一下各种情况。

首先,对于两个区间,我们用 [a1,a2][b1,b2] 表示在 AB 中的两个区间,那么什么情况下这两个区间没有交集呢:

只有这两种情况,写成代码的条件判断就是这样:

if b2 < a1 or a2 < b1:
    [a1,a2] 和 [b1,b2] 无交集

那么,什么情况下,两个区间存在交集呢?根据命题的否定,上面逻辑的否命题就是存在交集的条件:

# 不等号取反,or 也要变成 and
if b2 >= a1 and a2 >= b1:
    [a1,a2] 和 [b1,b2] 存在交集

接下来,两个区间存在交集的情况有哪些呢?穷举出来:

这很简单吧,就这四种情况而已。那么接下来思考,这几种情况下,交集是否有什么共同点呢?

我们惊奇地发现,交集区间是有规律的!如果交集区间是 [c1,c2],那么 c1=max(a1,b1)c2=min(a2,b2)!这一点就是寻找交集的核心,我们把代码更进一步:

while i < len(A) and j < len(B):
    a1, a2 = A[i][0], A[i][1]
    b1, b2 = B[j][0], B[j][1]
    if b2 >= a1 and a2 >= b1:
        res.append([max(a1, b1), min(a2, b2)])
    # ...

最后一步,我们的指针 ij 肯定要前进(递增)的,什么时候应该前进呢?

结合动画示例就很好理解了,是否前进,只取决于 a2b2 的大小关系:

while i < len(A) and j < len(B):
    # ...
    if b2 < a2:
        j += 1
    else:
        i += 1

代码

# A, B 形如 [[0,2],[5,10]...]
def intervalIntersection(A, B):
    i, j = 0, 0 # 双指针
    res = []
    while i < len(A) and j < len(B):
        a1, a2 = A[i][0], A[i][1]
        b1, b2 = B[j][0], B[j][1]
        # 两个区间存在交集
        if b2 >= a1 and a2 >= b1:
            # 计算出交集,加入 res
            res.append([max(a1, b1), min(a2, b2)])
        # 指针前进
        if b2 < a2: j += 1
        else:       i += 1
    return res

总结一下,区间类问题看起来都比较复杂,情况很多难以处理,但实际上通过观察各种不同情况之间的共性可以发现规律,用简洁的代码就能处理。

另外,区间问题没啥特别厉害的奇技淫巧,其操作也朴实无华,但其应用却十分广泛,接之前的几篇文章:

坚持原创高质量文章,致力于把算法问题讲清楚,欢迎关注我的公众号 labuladong 获取最新文章:

labuladong

上一篇:区间调度之区间合并问题

下一篇:信封嵌套问题

目录

Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Java
1
https://gitee.com/jjj201/fucking-algorithm.git
git@gitee.com:jjj201/fucking-algorithm.git
jjj201
fucking-algorithm
fucking-algorithm
master

搜索帮助