hdu4455(递推)

链接：点击打开链接

题意：给一个长度为n的序列，定义f(i,j)为区间[i,j]内中不同的数的个数，q次询问，每次询问x，求所有长度为x连续区间的f值之和

代码：

#include <set>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int a[1000005],f[1000005],id[1000005];
int dp[1000005],pos[1000005],sum[1000005];
int main(){
    int n,i,j,q,ans;
    while(scanf("%d",&n)!=EOF&&n){
        memset(pos,0,sizeof(pos));
        memset(sum,0,sizeof(sum));
        for(i=1;i<=n;i++){
            scanf("%d",&a[i]);
            sum[i-pos[a[i]]]++;                 //记录距离上一次出现的距离有几个数
            pos[a[i]]=i;
        }
        memset(f,0,sizeof(f));
        memset(id,0,sizeof(id));
        for(i=1;i<=n;i++){                      //从后数每个区间内有几种数
            if(id[a[n-i+1]]==0){                //不能用set，map会MLE
                f[i]=f[i-1]+1;
                id[a[n-i+1]]=1;
            }
            else
            f[i]=f[i-1];
        }
        ans=dp[1]=n;                            //主要在于不同长度之间的关系
        for(i=2;i<=n;i++){                      //每增加一个长度，则减少一个
            dp[i]=dp[i-1]-f[i-1];               //区间，并增加一部分数
            ans-=sum[i-1];
            dp[i]+=ans;                         //判断增加这部分数有几个能够符合条件
        }
        scanf("%d",&q);
        while(q--){
            scanf("%d",&ans);
            printf("%I64d\n",dp[ans]);
        }
    }
    return 0;
}