hdu4455 线性DP

最新推荐文章于 2017-10-23 21:31:12 发布

_OS

最新推荐文章于 2017-10-23 21:31:12 发布

阅读量581

点赞数

分类专栏： DP 文章标签： algorithm

本文链接：https://blog.csdn.net/ahjkl007/article/details/39342661

版权

DP 专栏收录该内容

31 篇文章 0 订阅

订阅专栏

题意：给定10^6的数字序列然后10^6个询问所长度为w的段中所有不相同的数字的个数的和是多少

解法：这一题很容易往数据结构上去想,蛋挞是个DP啊

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define ll long long
#define maxn 1000005
int pos[maxn],a[maxn],num[maxn],n,m,sum,vis[maxn];
ll dp[maxn],ex[maxn];
void _dp(){
    memset(vis,0,sizeof vis);
    ex[n+1]=0;
    for(int i=n;i>=1;i--){
        ex[i]=ex[i+1];
        if(!vis[a[i]]){
            vis[a[i]]=1,ex[i]++;
        }
    }
    sum=n-num[1];dp[1]=n;

    for(int i=2;i<=n;++i){
        dp[i]=dp[i-1]+sum-ex[n-i+2];
        sum-=num[i];
    }
}
int main(){
	int x;
	while(~scanf("%d",&n)){
        if(!n)break;
        memset(num,0,sizeof num);
        memset(pos,0,sizeof pos);
        for(int i=1;i<=n;++i){
            scanf("%d",&a[i]);
            num[i-pos[a[i]]]++;
            pos[a[i]]=i;
        }
        _dp();
		scanf("%d",&m);
        while(m--){
			scanf("%d",&x);
			printf("%I64d\n",dp[x]);
		}
	}
	return 0;
}