HDU 4455 Substring(DP)

如样例：
1 1 2 3 4 4 5；
明显dp[1]=n=7;

长度为1的时候有7个区间。

从长度为1到长度为2，就是把前6个区间往后增加一个数，最后一个元素不能构成长度为2的区间。

增加的6个数要看在该区间是否出现过，只要看它上一个相等的元素距离是否大于2
所以dp[2]=dp[1]-1+4;
 
以此类推就可以得出所以的dp值了。
dp[i]=dp[i-1]-A +B;
A是后缀不能得到i长度的元素的不同个数。

B是距离>=i元素的个数

#include<iostream>
#include<string>
#include<stdio.h>
#include<string.h>
#include<vector>
#include<math.h>
#include<queue>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
#define LL long long
const int INF=0x3f7f7f7f;
const int MAXN=1000005;
const double eps = 1e-10;
int cnt[MAXN],sum[MAXN],sum1[MAXN];
int a[MAXN],vis1[MAXN];
LL dp[MAXN];
int main()
{
    int i,n,w,q;
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0)break;
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
        }
        memset(cnt,0,sizeof(cnt));//cnt[d]:最长距离为d元素的个数
        memset(sum,0,sizeof(sum));//sum[id]:以id结尾的后缀不同元素个数
        memset(dp,0,sizeof(dp));//dp[w]:长度为w的答案
        memset(vis1,0,sizeof(vis1));
        memset(sum1,0,sizeof(sum1));//sum1[dis]:从左往右距离为dis的个数（dis=当前元素id-在之前此元素的位置）
        for(i=1;i<=n;i++)
        {
            int temp=vis1[a[i]];
            if(temp!=0)
            {
                cnt[i-temp]++;
            }
            else cnt[i]++;
            vis1[a[i]]=i;

        }

        for(i=n;i>=1;i--)
        {
            sum1[i]=sum1[i+1]+cnt[i];
        }//累计距离为i的个数
        memset(vis1,0,sizeof(vis1));
        for(i=n;i>=1;i--)
        {
            if(vis1[a[i]]==0)
            {
                sum[i]=sum[i+1]+1;
                vis1[a[i]]=1;
            }
            else sum[i]=sum[i+1];
        }
        dp[1]=n;
        for(i=2;i<=n;i++)
        {
            dp[i]=dp[i-1]-sum[n-i+2]+sum1[i];
        }
        scanf("%d",&q);
        while(q--)
        {
            scanf("%d",&w);
            printf("%lld\n",dp[w]);
        }
    }
    return 0;
}