英文作文 思维题

题目链接

想到了双指针的写法，看了题解之后感觉题解中的算法太妙了，记录一下
需要的是 $i<j$ 且 $s[i]=s[j]$ 且 $i$ 和 $j$ 之间的间隔不超过 $k$ ( $j-i-1$ $\le$ $k$ )
等价于 $s[i]=s[j]$ 且 $j-1$ $\ge$ i $\ge$ $j-k-1$
枚举 $j$
$mp[]$ 的定义是：$mp[s]$ 表示 $s$ 前面与 $s$ 间隔 $k$ 个之内 满足题设的个数
当计算到 $s[j]$ 的时候，$mp[]$ 中存放的是 $s[j-k-2]$ 到 $s[j-1]$ 中每个字符串出现的次数，因为$s[j-k-2]$ 对计算 $s[j]$ 的结果没有贡献，如果不清除其影响反而会影响结果，因此需要先消除其影响，直接在 $mp[]$ 中把 $s[j-k-2]--$ 即可，那么和当前字符串相同且间隔不超过 $k$ 且位于该字符串前面的字符串的个数为 $mp[s[i]]$，在后续的计算中，如果有字符串 $s[p]$ 和 $s[i]$ 相同，则 $s[p]$ 的结果需要多加一次 $s[i]$ 所造成的影响，因此$mp[s[i]]++$，后续也会同上一样被消去的

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int n, k;
const int N = 1e5 + 10;
string s[N];
map<string, int > mp;
int main(){
	freopen("in.in", "r", stdin);
	freopen("out.out", "w", stdout);
	ios::sync_with_stdio(false);
	cin >> n >> k;
	for (int i = 1; i <= n; i++){
		cin >> s[i];
	}
	ll ans = 0; 
	/* 
	 * mp[i] 表示 s 前面与 s 间隔 k 个之内 满足题设的个数
	 * 明显答案就是 mp[i]的累加和
	 * 当计算到 mp[i] 的时候 此时 mp[] 中存放的是 [i - k - 2, i - 1] 中每个字符串出现的次数
	 * 计算 mp[i] 不需要 mp[i - k - 2] 的值 需要去除它对答案的影响 因此 mp[i - k - 2]-- 将该字符串移除
	 * 此后 mp[i] 将影响后续 mp[i + 1, i + k + 1] 的答案 因此 mp[i]++ 表示该字符串在当前区间内多出现了一次
	 * 不得不说 这个算法设计的相当巧妙
	 * 
	*/
	for (int i = 1; i <= n; i++){
		if (i - k - 2 >= 1)mp[s[i - k - 2]]--;
		ans += mp[s[i]];
		mp[s[i]]++;
	}

	cout << ans << "\n";
	return 0;
}