LeetCode每日一题(699. Falling Squares)

wangjun861205

于 2023-02-18 11:01:31 发布

阅读量645

点赞数

分类专栏：数据结构算法文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/wangjun861205/article/details/129097985

版权

这篇博客详细介绍了LeetCode中的一道题目——Falling Squares。通过给出的示例和约束条件，解释了如何使用坐标压缩和线段树的数据结构来解决这道动态规划问题，旨在帮助读者理解解题思路和算法应用。

摘要由CSDN通过智能技术生成

There are several squares being dropped onto the X-axis of a 2D plane.

You are given a 2D integer array positions where positions[i] = [lefti, sideLengthi] represents the ith square with a side length of sideLengthi that is dropped with its left edge aligned with X-coordinate lefti.

Each square is dropped one at a time from a height above any landed squares. It then falls downward (negative Y direction) until it either lands on the top side of another square or on the X-axis. A square brushing the left/right side of another square does not count as landing on it. Once it lands, it freezes in place and cannot be moved.

After each square is dropped, you must record the height of the current tallest stack of squares.

Return an integer array ans where ans[i] represents the height described above after dropping the ith square.

Example 1:

Input: positions = [[1,2],[2,3],[6,1]]
Output: [2,5,5]

Explanation:
After the first drop, the tallest stack is square 1 with a height of 2.
After the second drop, the tallest stack is squares 1 and 2 with a height of 5.
After the third drop, the tallest stack is still squares 1 and 2 with a height of 5.
Thus, we return an answer of [2, 5, 5].

Example 2:

Input: positions = [[100,100],[200,100]]
Output: [100,100]

Explanation:
After the first drop, the tallest stack is square 1 with a height of 100.
After the second drop, the tallest stack is either square 1 or square 2, both with heights of 100.
Thus, we return an answer of [100, 100].
Note that square 2 only brushes the right side of square 1, which does not count as landing on it.

Constraints:

1 <= positions.length <= 1000
1 <= lefti <= 108
1 <= sideLengthi <= 106

用 coordinate compression 和 segment tree 来解


use std::collections::{
   HashMap, HashSet};

struct SegmentTree {
   
    n: usize,
    heights: Vec<i32>,
}

impl SegmentTree {
   
    fn new(n: usize) -> Self {
   
        let mut curr = 1;
        let mut total = 1;
        while curr < n {
   
            curr *= 2;
            total += curr;
        }
        Self {
    n: curr, heights: vec![0; total] }
    }

    fn query(