LeetCode第198场周赛题解

最新推荐文章于 2024-09-01 10:22:00 发布

IGP9

最新推荐文章于 2024-09-01 10:22:00 发布

阅读量204

点赞数

文章标签： leetcode

本文链接：https://blog.csdn.net/qq_44965626/article/details/107882235

版权

本文解析了LeetCode中的三道算法题：1. 换酒问题，通过模拟和递归解决；2. 子树标签相同节点计数，利用树形DP求解；3. 最多不重叠子字符串，采用前缀和和位运算优化。详细介绍了每题的思路和代码实现。

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

LeetCode第198场周赛题解

题目来源：LeetCode官网
本文主要内容为 Acwing y总视频讲解的听课笔记及个人理解

1.LeetCode 1518. 换酒问题

题目描述

小区便利店正在促销，用 numExchange 个空酒瓶可以兑换一瓶新酒。你购入了 numBottles 瓶酒。

如果喝掉了酒瓶中的酒，那么酒瓶就会变成空的。

请你计算最多能喝到多少瓶酒。

示例 1：

输入：numBottles = 9, numExchange = 3
输出：13
解释：你可以用 3 个空酒瓶兑换 1 瓶酒。
所以最多能喝到 9 + 3 + 1 = 13 瓶酒。

示例 2：

输入：numBottles = 15, numExchange = 4
输出：19
解释：你可以用 4 个空酒瓶兑换 1 瓶酒。
所以最多能喝到 15 + 3 + 1 = 19 瓶酒。

示例 3：
输入：numBottles = 5, numExchange = 5
输出：6
示例 4：
输入：numBottles = 2, numExchange = 3
输出：2

提示：
1 <= numBottles <= 100
2 <= numExchange <= 100

思路1：模拟

class Solution {
public:
    int numWaterBottles(int numBottles, int numExchange) {
        int res = numBottles;
        while(numBottles >= numExchange)
        {
            res += numBottles / numExchange;
            numBottles = numBottles / numExchange + numBottles % numExchange;
        }
        return res;
    }
};

思路2：递归

class Solution {
public:
    int res = 0;
    int numWaterBottles(int numBottles, int numExchange) {
        res = numBottles;
        dfs(numBottles, numExchange);
        return res;
    }
    void dfs(int a, int b)
    {
        if(a < b)return;
        res += a / b;
        dfs(a / b + a % b, b);
    }
};

2.LeetCode 1519. 子树中标签相同的节点数

题目描述

给你一棵树（即，一个连通的无环无向图），这棵树由编号从 0 到 n - 1 的 n 个节点组成，且恰好有 n - 1 条 edges 。树的根节点为节点 0 ，树上的每一个节点都有一个标签，也就是字符串 labels 中的一个小写字符（编号为 i 的节点的标签就是 labels[i] ）

边数组 edges 以 edges[i] = [ai, bi] 的形式给出，该格式表示节点 ai 和 bi 之间存在一条边。

返回一个大小为 n 的数组，其中 ans[i] 表示第 i 个节点的子树中与节点 i 标签相同的节点数。

树 T 中的子树是由 T 中的某个节点及其所有后代节点组成的树。

示例 1：

输入：n = 7, edges = [[0,1],[0,2],[1,4],[1,5],[2,3],[2,6]], labels = “abaedcd”
输出：[2,1,1,1,1,1,1]
解释：节点 0 的标签为 ‘a’ ，以 ‘a’ 为根节点的子树中，节点 2 的标签也是 ‘a’ ，因此答案为 2 。注意树中的每个节点都是这棵子树的一部分。
节点 1 的标签为 ‘b’ ，节点 1 的子树包含节点 1、4 和 5，但是节点 4、5 的标签与节点 1 不同，故而答案为 1（即，该节点本身）。

提示：
1 <= n <= 10^5
edges.length == n - 1
edges[i].length == 2
0 <= ai, bi < n
ai != bi
labels.length == n
labels 仅由小写英文字母组成

分析

算法：树形DP—o(26n)
思路：

用一个二维数组f来存储以当前节点为根节点的树的26个字母的数量
dfs搜索整棵树，用所有以当前节点的子节点为根的子树26个字母的数量来更新以当前节点为根的树的26个字母的数量，记录与当前节点字母相同的树的字母的数量

代码

class Solution {
public:
    vector<int>ans;
    vector<vector<int>>g;
    vector<vector<int>>cnt;
    string labels;
    vector<int> countSubTrees(int n, vector<vector<int>>& edges, string _labels) {
        ans = vector<int>(n), g = vector<vector<int>>(n);
        cnt = vector<vector<int>>(n, vector<int>(26, 0));
        labels = _labels;
        for(auto e : edges)
        {
            int a = e[0], b = e[1];
            g[a].push_back(b);
            g[b].push_back(a);
        }
        dfs(0, -1);
        return ans;
    }
    void dfs(int u, int fa)
    {
        cnt[u][labels[u] - 'a']++;
        for(auto e : g[u])
        {
            int son = e;
            if(son == fa)continue;
            dfs(son, u);
            for(int i = 0; i < 26; i++)
                cnt[u][i] += cnt[son][i];
        }
        ans[u] = cnt[u][labels[u] - 'a'];
    }
};

3.LeetCode 1520. 最多的不重叠子字符串

题目描述

给你一个只包含小写字母的字符串 s ，你需要找到 s 中最多数目的非空子字符串，满足如下条件：
这些字符串之间互不重叠，也就是说对于任意两个子字符串 s[i…j] 和 s[k…l] ，要么 j < k 要么 i > l 。
如果一个子字符串包含字符 char ，那么 s 中所有 char 字符都应该在这个子字符串中。
请你找到满足上述条件的最多子字符串数目。如果有多个解法有相同的子字符串数目，请返回这些子字符串总长度最小的一个解。可以证明最小总长度解是唯一的。

请注意，你可以以任意顺序返回最优解的子字符串。

示例 1：
输入：s = “adefaddaccc”
输出：[“e”,“f”,“ccc”]
解释：下面为所有满足第二个条件的子字符串：
[
“adefaddaccc”
“adefadda”,
“ef”,
“e”,
“f”,
“ccc”,
]
如果我们选择第一个字符串，那么我们无法再选择其他任何字符串，所以答案为 1 。如果我们选择 “adefadda” ，剩下子字符串中我们只可以选择 “ccc” ，它是唯一不重叠的子字符串，所以答案为 2 。同时我们可以发现，选择 “ef” 不是最优的，因为它可以被拆分成 2 个子字符串。所以最优解是选择 [“e”,“f”,“ccc”] ，答案为 3 。不存在别的相同数目子字符串解。

示例 2：
输入：s = “abbaccd”
输出：[“d”,“bb”,“cc”]
解释：注意到解 [“d”,“abba”,“cc”] 答案也为 3 ，但它不是最优解，因为它的总长度更长。

提示：
1 <= s.length <= 10^5
s 只包含小写英文字母。

分析

思路：

找到包含某个小写字母的最短合法区间（不会超过26个），这些区间一定是最优解
1.初始化每个字母出现次数的前缀和数组
2.找到只包含某个字母的的最短区间
3.迭代，若某个字母a区间包含其他字母b，则合并a、b两个区间，直到无法合并为止
从这26个合法区间中挑选数量最多最短且满足题目要求的区间作为答案
1.这些合法区间不可能相交，相交的话则可以把它缩短，这就与它是最优解相矛盾
2.双重循环遍历所有区间，若一个区间x是另一个区间y的真子集，则舍弃区间y
3.遍历所有区间，将没被舍弃的区间放入答案数组返回

代码

class Solution {
public:
    vector<string> maxNumOfSubstrings(string s) {
        int n = s.size();
        s = ' ' + s;
        vector<vector<int>>seg(26, {INT_MAX, INT_MIN});
        vector<vector<int>>sum(26, vector<int>(n + 1, 0));
        //初始胡：初始化每个字母出现次数的前缀和数组和横跨的区间
        for(int i = 1; i <= n; i++)
        {
            int t = s[i] - 'a';
            for(int j = 0; j < 26; j++)
            {
                sum[j][i] = sum[j][i - 1];
                if(t == j)sum[t][i]++;
            }
            seg[t][0] = min(seg[t][0], i);
            seg[t][1] = max(seg[t][1], i);
        }
        
		//合并区间：若当前字母i区间中包含其他字母j，则把i、j两个区间合并
        bool flag = true;
        while(flag)
        {
            flag = false;
            for(int i = 0; i < 26; i++)
            {
                auto cur = seg[i];
                if(cur[0] > cur[1])continue;
                for(int j = 0; j < 26; j++)
                    if(i != j && sum[j][cur[1]] - sum[j][cur[0] - 1])
                    {
                        cur[0] = min(cur[0], seg[j][0]);
                        cur[1]=  max(cur[1], seg[j][1]);
                    }
                if(cur != seg[i])
                {
                    seg[i] = cur;
                    flag = true;
                }
            }
        }
        //删去一些区间内包含其他区间的区间
        for(int i = 0; i < 26; i++)
            for(int j = 0; j < 26; j++)
                if(i != j && seg[j][0] != INT_MAX && seg[j][0] >= seg[i][0] && seg[j][1] <= seg[i][1])
                {
                    seg[i][0] = INT_MAX, seg[i][1] = INT_MIN;
                    break;
                }
         //将剩下的合法区间作为答案输出
        vector<string>res;
        for(int i = 0; i < 26; i++)
            if(seg[i][0] <= seg[i][1])
                res.push_back(s.substr(seg[i][0], seg[i][1] - seg[i][0] + 1));
        return res;
    }
};

4.LeetCode 1521. 找到最接近目标值的函数值

题目描述

Winston 构造了一个如上所示的函数 func 。他有一个整数数组 arr 和一个整数 target ，他想找到让 |func(arr, l, r) - target| 最小的 l 和 r 。

请你返回 |func(arr, l, r) - target| 的最小值。

请注意， func 的输入参数 l 和 r 需要满足 0 <= l, r < arr.length 。

示例 1：
输入：arr = [9,12,3,7,15], target = 5
输出：2
解释：所有可能的 [l,r] 数对包括 [[0,0],[1,1],[2,2],[3,3],[4,4],[0,1],[1,2],[2,3],[3,4],[0,2],[1,3],[2,4],[0,3],[1,4],[0,4]]， Winston 得到的相应结果为 [9,12,3,7,15,8,0,3,7,0,0,3,0,0,0] 。最接近 5 的值是 7 和 3，所以最小差值为 2 。
示例 2：
输入：arr = [1000000,1000000,1000000], target = 1
输出：999999
解释：Winston 输入函数的所有可能 [l,r] 数对得到的函数值都为 1000000 ，所以最小差值为 999999 。
示例 3：
输入：arr = [1,2,4,8,16], target = 0
输出：0

提示：
1 <= arr.length <= 10^5
1 <= arr[i] <= 10^6
0 <= target <= 10^7

分析

算法：前缀和+位运算+双指针—o(20n)
思路：

暴力来做三层循环o(n³)
优化第三层循环，求一段区间所有数与起来的值，开一个20位的前缀和数组存储区间每位0的个数，按位遍历当前区间各位0的个数可以求得当前区间所有数与起来的值
用双指针优化双重循环枚举，a & x <= a，当固定i时，j取val[j, i] <= target的j的最右值，当i往后移时，[j, i]一定会减小，所以j也要往后移

代码

class Solution {
public:
    vector<vector<int>>s;
    int get_sum(int l, int r)
    {
        int res = 0;
        for(int i = 0; i < 20; i++)
            if(s[i][r] - s[i][l - 1] == 0)
                res += 1 << i;
        return res;
    }
    int closestToTarget(vector<int>& arr, int target) {
        int n = arr.size();
        s = vector<vector<int>>(20, vector<int>(n + 1));
        for(int i = 0; i < 20; i++)
            for(int j = 1; j <= n; j++)
            {
                s[i][j] = s[i][j - 1];
                if(!(arr[j - 1] >> i & 1))s[i][j]++;
            }
        int res = INT_MAX;
        for(int i = 1, j = 1; i <= n; i++)
        {
            while(j + 1 <= i && get_sum(j + 1, i) <= target)j++;
            res = min(res, abs(target - get_sum(j, i)));
            if(j < i)res = min(res, abs(target - get_sum(j + 1, i)));
        }
        return res;
    }
};