HDU4455(DP)

这道题思路十分巧妙，我觉得看到10^6次方的数据和查询，一般都会先往线段树方向考虑。我也想了半天，实在思考不出来去看了邝大神的博客才恍然大悟。编程实现不难主要是思路很难想到。时间复杂度为O（n），dp方程:dp[i] = dp[i-1]+A-B (A,B含义见注释)。

代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#define maxn 1000100
#define ll long long
using namespace std;

int A[maxn], B[maxn], C[maxn], num[maxn], flag[maxn];
int n, q;
ll dp[maxn];

void init()
{
    //初始化A数组表示最后i-1个有多少个不重复的元素
    memset(flag, 0, sizeof flag);
    A[1] = 0;
    for(int i=2; i<=n; i++){
        A[i] = A[i-1];
        if(flag[num[n-i+1]]==0){
            flag[num[n-i+1]] = 1;
            A[i] ++;
        }
    }
    //初始化B数组表示最近的相同元素相隔的距离为i的个数
    memset(flag, -1, sizeof flag);
    memset(B, 0, sizeof B);
    for(int i=0; i<n; i++){
        if(flag[num[i]]==-1){
            flag[num[i]] = i;
            B[i+1]++;
        }
        else{
            B[i-flag[num[i]]] ++;
            flag[num[i]] = i;
        }
    }
}

void solve()
{
    init();
    int sum;
    dp[1] = sum = n;
    for(int i=2; i<=n; i++){
        sum-=B[i-1];
        dp[i] = dp[i-1]-A[i]+sum;
    }
}

int main()
{
//    freopen("in.txt", "r", stdin);

    int tq;
    while(scanf("%d", &n)!=EOF && n)
    {
        for(int i=0; i<n; i++){
            scanf("%d", &num[i]);
        }
        scanf("%d", &q);
        solve();
        for(int i=0; i<q; i++){
            scanf("%d", &tq);
            printf("%I64d
", dp[tq]);
        }
    }
    return 0;
}