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;
}
posted @ 2021-08-10 21:48  5k_sync_closer  阅读(0)  评论(0编辑  收藏  举报  来源