在写英语作文的时候,两个相同单词靠的太近肯定不好。现在 ZHR 给了你一段n个单词的英文,问你有多少对相同单词中间间隔的单词数小于等于k 。
基本思想:利用map的映射将字符串映射为数字保存,结合双指针算法降低时间复杂度(本题n的范围到1e5,暴力肯定超时)代码实现如下
#include <bits/stdc++.h>
using namespace std;
string str[100010];
int n, k;
map<string, int>cnt;
//利用map将字符映射为数字;
int main()
{
cin >> n >> k;
for (int i = 1; i <= n; i++)
cin >> str[i];
int j = 1, ans = 0;
//定义指针j;
for (int i = 1; i <= n; i++)
{
while (i - j - 1 > k)//确保区间满足题意,即i和j之间相差不超过k;
{
cnt[str[j]]--;
j++;//右指针向前走,左指针同样也要向前走,同时把cnt[]储存的str[j]删去;
}
ans += cnt[str[i]];//若当前区间中有超过一对的str[i],把它加入答案即可,
cnt[str[i]]++;//更新map中str[i]的数量;
}
cout << ans;
return 0;
}