用尺取法针对“字符串序列,求无重复子串的最大长度”之类题目较多,针对数值的相对较少。刚好小孩参加的比赛有这个类型,稍加修改还是可以通过的。
********************************************************************
Description:
小J有N个糖果摆成一列,每个糖果有不同的颜色
小J可以吃连续的K个糖果
现在他希望这K个糖果所包含的糖果的颜色尽可能多
请你求出来。
Format:
Input
一行给出数字N,K
第二行N个数字,代表每个糖果的颜色,其值<=1e9
N,K <=3e5
Output
针对每个询问,输出结果。
AC通过的C++代码:
#include <bits/stdc++.h>
using namespace std;
int main() { //小J吃糖果
int N, K;
cin >> N >> K;
vector<int> candies(N);
for (int i = 0; i < N; ++i) {
cin >> candies[i];
}
int maxDistinct = 0; // 存储最多颜色的数量
int distinctCount = 0; // 当前窗口内不同颜色的数量
unordered_map<int, int> colorFreq; // 存储颜色及其频率
int left = 0, right = 0; // 窗口的左右指针
while (right < N) {
// 如果当前颜色在窗口内首次出现,增加不同颜色计数
if (colorFreq[candies[right]] == 0) {
distinctCount++;
}
colorFreq[candies[right]]++; // 更新颜色频率
right++; // 将右指针向右移动
if (right - left == K) { // 如果窗口大小达到K,需要调整左指针
// 如果移除的颜色是唯一的(即窗口内只有一个这种颜色)
if (colorFreq[candies[left]] == 1) {
maxDistinct = max(maxDistinct, distinctCount);// 更新当前最大不同颜色数
distinctCount--;//减少不同颜色计数
}
colorFreq[candies[left]]--; // 减少左指针指向的颜色的频率
left++;// 将左指针向右移动
maxDistinct = max(maxDistinct, distinctCount);// 更新最大不同颜色数
}
}
cout << maxDistinct << endl;// 输出结果
return 0;
}