CF368B Sereja and Suffixes
y1s1,这题评红感觉低了。
- 没思路?提醒三个字:前缀和!
- 还是没思路?再提醒三个字:倒着推!
- 到现在还没思路?往下看吧。
思路
这道题采用(类似)前缀和的方法优化。
设 $s[i]$ 为 $[i,n]$ 区间内有几个不同数。
现在只需要推出 $s$, 处理就可以 $O(1)$ 了。
这里不能正着推,想想都知道前缀和是从 1 开始的啊。
现在思考一个问题: $s[i]$ 与 $s[i+1]$ 有什么关系?
答:如果 $a[i]$ 在后面出现过,则 $s[i]=s[i+1]$,否则 $s[i]=s[i+1]+1$。
那么推法就显而易见了,可以用个 hash 维护是否出现,但 $10^5$ 没必要,数组就行。
初始状态:显而易见 $[n,n]$ 里只有 1 个数,所以 $s[n]=1$。
然后 $a[n]$ 就算见过了,所以 $vis[a[n]]=1$。
代码
#include <iostream>
using namespace std;
int n, m, s[100001], a[100001], l;bool vis[100001];
int main()
{
cin >> n >> m;
for(int i = 1;i <= n;++i)
cin >> a[i];
s[n] = 1;vis[a[n]] = 1;
for(int i = n - 1;i >= 1;--i)
{
s[i] = s[i + 1];
if(!vis[a[i]]) ++s[i];
vis[a[i]] = 1;
}
for(int i = 0;i < m;++i)
cin >> l, cout << s[l] << endl;
return 0;
}