题目链接:
http://acm.hdu.edu.cn/showproblem.php?pid=4455
题目意思:
给一个长度为n的数组,有q个询问w,输出区间长度为w的所有子序列中不同数的个数总和。
解题思路:
dp+预处理优化,o(n)的时间复杂度才能过。
dp[i]表示子串长度
对于样例
长度为1:1 1 2 3 4 4 5
长度为2:11 12 23 34 44 45 dp[2]在dp[1]的基础上去掉了最后一个5,然后判断该长度序列最后数字是否再前面出现。没有出现则加一,否则不变。注意此时前面长度少于当前长度的也要减去。
先预处理出cnt[i]:表示距离为i的两相邻相同数k的组数,如果某数的出现相邻位置分别为k1,k2,如果k2-k1+1>i,则需增加1。len[i]表示区间i~n不同数的个数。
dp[i]=dp[i-1]-len[n-i+2]+(除去cnt[j],j<=k的个数,也就是新增加的个数)。
代码:
#include<iostream>
#include<cmath>
#include<cstdio>
#include<sstream>
#include<cstdlib>
#include<string>
#include<cstring>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<ctime>
#include<bitset>
#define eps 1e-6
#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define ll __int64
#define LL long long
#define lson l,m,(rt<<1)
#define rson m+1,r,(rt<<1)|1
#define M 1000000007
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
#define Maxn 1100000
int la[Maxn],len[Maxn];
int cnt[Maxn];
int sa[Maxn];
ll dp[Maxn];
int main()
{
int n,m;
while(scanf("%d",&n)&&n)
{
for(int i=1;i<=n;i++)
scanf("%d",&sa[i]);
memset(cnt,0,sizeof(cnt));
memset(la,-1,sizeof(la)); //注意la是代表的sa[i]出现的最迟位置,
//小标取值范围不一定为0~n,而是0~10^6.
la[sa[n]]=n;//标记sa[n]出现的最迟位置
len[n]=1;
for(int i=n-1;i>=1;i--)
{
if(la[sa[i]]==-1) //之前没有出现
len[i]=len[i+1]+1; //个数+1
else
len[i]=len[i+1]; //个数不变
la[sa[i]]=i; //出现了
}
memset(la,0,sizeof(la));
for(int i=1;i<=n;i++)
{
cnt[i-la[sa[i]]+1]++; //统计相同的距离。注意之前没出现的距离为i.
la[sa[i]]=i;
}
dp[1]=n;
int s=n-cnt[1];//总共才n个区间,以最后一个数来统计。减去距离为1的
for(int i=2;i<=n;i++)
{
s-=cnt[i]; //减去距离为i的
dp[i]=dp[i-1]-len[n-i+2]+s;//减去最后丢掉的,加上新添加的
}
int m;
scanf("%d",&m);
while(m--)
{
int a;
scanf("%d",&a);
printf("%I64d\n",dp[a]);
}
}
return 0;
}