Q1 找到两个数组中的公共元素
-
解题思路:
- 分别用两个哈希表记录两个数组中各元素的出现次数
- 遍历其中一个哈希表,如果对应元素在两个哈希表中均出现,分别将其出现次数加到答案不同位置
-
解题代码:
class Solution {
public:
vector<int> findIntersectionValues(vector<int>& nums1, vector<int>& nums2) {
unordered_map<int, int> um1, um2;
for(auto num : nums1)
um1[num] += 1;
for(auto num : nums2)
um2[num] += 1;
vector<int> ans(2, 0);
for(auto [v, c] : um2)
{
if(um1.count(v))
{
ans[0] += um1[v];
ans[1] += c;
}
}
return ans;
}
};
Q2 消除相邻近似相等字符
-
解题思路:
- 从左向右遍历字符串
- 检查到某位置与其左侧的字符近似相等:
- 需要将该字符进行修改
- 一定可以找到一个修改后的字符使其既不与左侧字符近似相等,也不与右侧字符近似相等
- 因此下一个检查位置直接是向右移动两个字符的位置
-
解题代码:
class Solution {
public:
int removeAlmostEqualCharacters(string word) {
int n = word.size();
int ans = 0;
for(int i = 1; i < n; i++)
{
if(abs((int)word[i] - (int)word[i-1]) <= 1)
{
ans += 1;
i += 1;
}
}
return ans;
}
};
Q3 最多K个重复元素的最长子数组
-
解题思路:
- 滑动窗口
- 使用两个哈希表,一个记录窗口中各元素出现了多少次,另一个记录窗口中各出现次数出现了多少次
- 当窗口中的最大出现次数大于k了,右移左指针
-
解题代码:
class Solution {
public:
int maxSubarrayLength(vector<int>& nums, int k) {
unordered_map<int, int> um;
map<int, int> cnts;
int n = nums.size();
int left = 0, right = 0;
int ans = 0;
while(right < n)
{
int pre = um[nums[right]];
if(pre != 0)
cnts[pre] -= 1;
if(cnts[pre] == 0)
cnts.erase(pre);
pre += 1;
um[nums[right]] += 1;
cnts[pre] += 1;
while(left <= right && cnts.rbegin()->first > k)
{
int pre = um[nums[left]];
cnts[pre] -= 1;
if(cnts[pre] == 0)
cnts.erase(pre);
pre -= 1;
um[nums[left]] -= 1;
if(pre != 0)
cnts[pre] += 1;
left += 1;
}
ans = max(ans, right-left+1);
right += 1;
}
return ans;
}
};
Q4 关闭分部的可行集合数目
-
解题思路:
- 二进制枚举 + Floyid
- 枚举所有可能的打开分部的情况,用一个n位的二进制数表示,0表示关闭,1表示打开
- 判断在每一种情况下,任意一对打开的分部之间的距离,是否都满足不超过maxDistance,如果是,则更新答案
-
解题代码:
class Solution {
public:
int numberOfSets(int n, int maxDistance, vector<vector<int>>& roads) {
int mask = 1 << n;
int ans = 0;
for(int j = 0; j < mask; j++)
{
vector<int> vis(n, false);
for(int i = 0; i < n; i++)
{
if((j >> i) & 1)
vis[i] = true;
}
vector<vector<long long>> f(n, vector<long long>(n, INT_MAX));
//初始化
for(auto &road : roads)
{
int x = road[0];
int y = road[1];
long long w = road[2];
if(vis[x] && vis[y])
{
f[x][y] = min(f[x][y], w);
f[y][x] = min(f[y][x], w);
}
}
for(int i = 0; i < n; i++)
f[i][i] = 0;
//Floyid
for(int k = 0; k < n; k++)
{
for(int i = 0; i < n; i++)
{
for(int j = 0; j < n; j++)
f[i][j] = min(f[i][j], f[i][k] + f[k][j]);
}
}
//判断
bool state = true;
for(int i = 0; i < n; i++)
{
for(int j = 0; j < n; j++)
{
if(vis[i] && vis[j] && f[i][j] > maxDistance)
state = false;
}
}
if(state)
ans += 1;
}
return ans;
}
};