本文涉及的基础知识点

C++二分查找

LeetCode2055. 蜡烛之间的盘子

给你一个长桌子，桌子上盘子和蜡烛排成一列。给你一个下标从 0 开始的字符串 s ，它只包含字符 ‘’ 和 ‘|’ ，其中 '’ 表示一个 盘子 ，‘|’ 表示一支 蜡烛 。
同时给你一个下标从 0 开始的二维整数数组 queries ，其中 queries[i] = [lefti, righti] 表示 子字符串 s[lefti…righti] （包含左右端点的字符）。对于每个查询，你需要找到 子字符串中 在 两支蜡烛之间 的盘子的 数目 。如果一个盘子在 子字符串中 左边和右边 都 至少有一支蜡烛，那么这个盘子满足在 两支蜡烛之间 。
比方说，s = “|||||" ，查询 [3, 8] ，表示的是子字符串 "||**|” 。子字符串中在两支蜡烛之间的盘子数目为 2 ，子字符串中右边两个盘子在它们左边和右边 都 至少有一支蜡烛。
请你返回一个整数数组 answer ，其中 answer[i] 是第 i 个查询的答案。
示例 1:

输入：s = “||***|”, queries = [[2,5],[5,9]]
输出：[2,3]
解释：

• queries[0] 有两个盘子在蜡烛之间。
• queries[1] 有三个盘子在蜡烛之间。
  示例 2:

输入：s = “||||||*”, queries = [[1,17],[4,5],[14,17],[5,11],[15,16]]
输出：[9,0,0,0,0]
解释：

• queries[0] 有 9 个盘子在蜡烛之间。
• 另一个查询没有盘子在蜡烛之间。
  提示：
  3 <= s.length <= 10⁵
  s 只包含字符 ‘*’ 和 ‘|’ 。
  1 <= queries.length <= 10⁵
  queries[i].length == 2
  0 <= lefti <= righti < s.length

二分查找

将蜡烛的下标按升序放到v1中，将盘子的下标按升序放到v2中。
对于每个查找，现在v1中查询[left,right+1)区间的蜡烛，如果蜡烛数小于2，返回0。否则，令第一个蜡烛为i1，最后一个蜡烛为i2。
在v2中查询区间[i1+1,i2)的数量。

代码

核心代码

class Solution {
		public:
			vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
				vector<int> v1, v2;
				for (int i = 0; i < s.length(); i++) {
					if ('|' == s[i]) {
						v1.emplace_back(i);
					}
					else {
						v2.emplace_back(i);
					}
				}
				vector<int> ret;
				for (const auto& v : queries) {
					auto it1 = lower_bound(v1.begin(), v1.end(), v[0]);
					auto it2 =  upper_bound(v1.begin(), v1.end(), v[1]);
					if (it2 - it1 < 2) { ret.emplace_back(0); continue; }
					--it2;
					auto it3 = lower_bound(v2.begin(), v2.end(), *it1);
					auto it4 = lower_bound(v2.begin(), v2.end(), *it2);
					ret.emplace_back(it4-it3);
				}
				return ret;
			}
		};

单元测试

string s;
		vector<vector<int>> queries;
		TEST_METHOD(TestMethod11)
		{
			s = "**|**|***|", queries = { {2,5},{5,9} };
			auto res = Solution().platesBetweenCandles(s, queries);
			AssertEx({ 2,3 }, res);
		}
		TEST_METHOD(TestMethod12)
		{
			s = "***|**|*****|**||**|*", queries = { {1,17},{4,5},{14,17},{5,11},{15,16} };
			auto res = Solution().platesBetweenCandles(s, queries);
			AssertEx({ 9,0,0,0,0 }, res);
		}
		TEST_METHOD(TestMethod3)
		{
			s = "||*", queries = { {2,2} };
			auto res = Solution().platesBetweenCandles(s, queries);
			AssertEx({ 0 }, res);
		}