【C++差分数组】2381. 字母移位 II|1793

news2024/10/5 8:17:12

本文涉及知识点

C++差分数组

LeetCode2381. 字母移位 II

给你一个小写英文字母组成的字符串 s 和一个二维整数数组 shifts ，其中 shifts[i] = [starti, endi, directioni] 。对于每个 i ，将 s 中从下标 starti 到下标 endi （两者都包含）所有字符都进行移位运算，如果 directioni = 1 将字符向后移位，如果 directioni = 0 将字符向前移位。
将一个字符向后移位的意思是将这个字符用字母表中下一个字母替换（字母表视为环绕的，所以 ‘z’ 变成 ‘a’）。类似的，将一个字符向前移位的意思是将这个字符用字母表中前一个字母替换（字母表是环绕的，所以 ‘a’ 变成 ‘z’ ）。
请你返回对 s 进行所有移位操作以后得到的最终字符串。
示例 1：
输入：s = “abc”, shifts = [[0,1,0],[1,2,1],[0,2,1]]
输出：“ace”
解释：首先，将下标从 0 到 1 的字母向前移位，得到 s = “zac” 。
然后，将下标从 1 到 2 的字母向后移位，得到 s = “zbd” 。
最后，将下标从 0 到 2 的字符向后移位，得到 s = “ace” 。
示例 2:
输入：s = “dztz”, shifts = [[0,0,0],[1,1,1]]
输出：“catz”
解释：首先，将下标从 0 到 0 的字母向前移位，得到 s = “cztz” 。
最后，将下标从 1 到 1 的字符向后移位，得到 s = “catz” 。
提示：
1 <= s.length, shifts.length <= 5 * 10⁴
shifts[i].length == 3
0 <= starti <= endi < s.length
0 <= directioni <= 1
s 只包含小写英文字母。

差分数组

将directioni的0全部改成-1。
对各 shifts[i]，vDiff[starti]++,vDiff[endi+1]–。
令diff对应的数据数组为a[i]
tmp = (s[i]-‘a’+a[i])%26;注意：tmp可能是负数。
则s[i] = ‘a’ + (tmp+26)%26;

代码

核心代码

class Solution {
		public:
			string shiftingLetters(string s, vector<vector<int>>& shifts) {
				const int N = s.length();
				vector<int> diff(N + 1);
				for (const auto& v : shifts) {
					const int iAdd = (v[2] == 1) ? 1 : -1;
					diff[v[0]]+= iAdd;
					diff[v[1] + 1]-=iAdd;
				}
				int cur = 0;
				string ret;
				for (int i = 0; i < N;i++) {
					cur += diff[i];
					const auto tmp = (s[i] - 'a' + cur) % 26;
					ret += char('a' + (tmp+26)%26);
				}
				return ret;
			}
		};

单元测试

string s;
		vector<vector<int>> shifts;
		TEST_METHOD(TestMethod1)
		{
			s = "abc", shifts = { {0,1,0} };
			auto res = Solution().shiftingLetters(s, shifts);
			AssertEx(string("zac"), res);
		}
		TEST_METHOD(TestMethod11)
		{
			s = "abc", shifts = { {0,1,0},{1,2,1},{0,2,1} };
			auto res = Solution().shiftingLetters(s, shifts);
			AssertEx(string("ace"), res);
		}
		TEST_METHOD(TestMethod12)
		{
			s = "dztz", shifts = { {0,0,0},{1,1,1} };
			auto res = Solution().shiftingLetters(s, shifts);
			AssertEx(string("catz"), res);
		}

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标及时的反馈拉伸区（难度合适）专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功成功+反思=成功