题目列表
3090. 每个字符最多出现两次的最长子字符串
3091. 执行操作使数据元素之和大于等于 K
3092. 最高频率的 ID
3093. 最长公共后缀查询
一、每个字符最多出现两次的最长子字符串
非常经典的滑动窗口问题,即动态维护一段区间,使得这段区间满足题目要求,代码如下
class Solution {
public:
int maximumLengthSubstring(string s) {
int cnt[26] = { 0 };
int n = s.size();
int ans = 0;
for(int l = 0, r = 0; r < n; r++){
cnt[s[r]-'a']++;
while(cnt[s[r]-'a']>2){
cnt[s[l]-'a']--;
l++;
}
ans = max(r-l+1,ans);
}
return ans;
}
};
二、执行操作使数据元素之和大于等于K
题目中有 将元素+1 和 复制元素 两个操作。我们先来想想这两个操作哪个先执行,结果会更优?很显然,先+1再x2得到的结果 比 先x2再+1得到的结果 更大,即我们应该优先进行+1操作,然后开始复制得到的数最大,但是我们要加到多大再进行复制需要的操作次数才最小呢?我们可以暴力枚举出所有情况,然后取得最大值,代码如下
class Solution {
public:
int minOperations(int k) {
int ans = INT_MAX;
for(int i=1;i<=k;i++){
// 进行 i-1 次 +1操作
// 再进行 ceil((k-i)/i) (ceil---向上取整) 次 复制操作
ans=min(ans,i-1+(k-i+i-1)/i);
}
return ans;
}
};
那有没有更加优雅的做法呢?
我们来看看i-1+(k-1)/i这个表达式,用x替换成i,得到 f(x) = x-1+(k-1)/x,问如何取f(x)的最小值?这个高中都学过,对勾函数,当x=sqrt(k-1)时,f(x)最小(如果不知道对勾函数,对它进行求导找最小值也是可以的)这里我们的x要取整数。代码如下
class Solution {
public:
int minOperations(int k) {
// 进行 x-1 次 +1操作
// 再进行 ceil((k-x)/x) (ceil---向上取整) 次 复制操作
// f(x) = x-1+(k-x+x-1)/x = x-1+(k-1)/x
if(k==1) return 0; // 分母不能为零,要特判
int x = sqrt(k-1);
if(x*x==k-1) return x*2-1;
else return min(x-1+(k-1)/x,x+1-1+(k-1)/(x+1));
}
};
三、最高频率的ID
本题就是要求我们在统计ID出现次数的基础上维护出现频率的最大值 ,统计ID出现次数很简单,我们可以用哈希表存储,关键是如何维护出现频率的最大值,可以用multiset来帮助我们维护最大值,代码如下
// 哈希表 + multiset
class Solution {
public:
vector<long long> mostFrequentIDs(vector<int>& nums, vector<int>& freq) {
int n = nums.size();
vector<long long> ans(n);
multiset<long long> s;
unordered_map<int,long long> mp;
for(int i = 0; i < n; i++){
int x = nums[i];
auto it = s.find(mp[x]);
if(it != s.end())
s.erase(it);
mp[x]+=freq[i];
s.insert(mp[x]);
ans[i]=*s.rbegin();
}
return ans;
}
};
// 哈希表 + 堆(懒删除) --- 可以了解一下
// 懒删除:当我们要取出堆顶元素的时候,看它的ID和freq是否匹配,
// 如果不匹配说明已经被修改,该数据无效 pop,如果相同,则是答案
class Solution {
public:
vector<long long> mostFrequentIDs(vector<int>& nums, vector<int>& freq) {
int n = nums.size();
vector<long long>ans(n);
priority_queue<pair<long long,int>> pq;//大堆<freq,ID>
unordered_map<int,long long>mp;
for(int i=0;i<n;i++){
mp[nums[i]] += freq[i];
pq.emplace(mp[nums[i]],nums[i]);
while(pq.top().first!=mp[pq.top().second])
pq.pop();
ans[i]=pq.top().first;
}
return ans;
}
};
四、最长公共后缀查询
这题没啥可说的,之前周赛出过类似的,用字典树来做
代码如下
struct Node{
Node* child[26]={0};
int idx = -1;// 记录当前后缀下长度最短的字符串下标
};
class Solution {
public:
vector<int> stringIndices(vector<string>& wordsContainer, vector<string>& wordsQuery) {
// 建立字典树
int pos = -1, mn = INT_MAX;
Node*root = new Node;
for(int i = 0; i < wordsContainer.size(); i++){
const string &s = wordsContainer[i];
int n = s.size();
if(mn > n) pos = i, mn = n; // 找最短的字符串下标
Node* cur = root;
for(int j = n-1; j >= 0; j--){
int x = s[j] - 'a';
if(cur->child[x] == nullptr)
cur->child[x] = new Node;
cur = cur->child[x];
if(cur->idx == -1 || wordsContainer[cur->idx].size() > n)
cur->idx = i;
}
}
// 查询
int m = wordsQuery.size();
vector<int> ans(wordsQuery.size(),-1);
for(int i = 0; i < m; i++){
const string &s = wordsQuery[i];
int n = s.size();
Node* cur = root;
for(int j = n-1; j >= 0; j--){
int x = s[j] - 'a';
cur = cur->child[x];
if(cur) ans[i] = cur->idx;
else break;
}
if(ans[i] == -1) ans[i] = pos;
}
return ans;
}
};