好久没写博客了,看了下上一篇博客还是八月份的最后一天写的,然后就因为阿里拥抱变化,忙着找工作去了;9月份找到工作,10月份又胡思乱想了一个月,但其实都是没必要的,应该明白胡思乱想解决不了任何问题的,实在得不到的又何必强求,顺其自然吧。加油吧,少年!!!
题目链接:点击进入
下午训练的时候这道题想了好几个小时还是完全没有思路,但是静下心来还是可以看到一些规律的。首先w=i+1的情况和w=i时的情况是有关系的,具体的我们设dp[i]表示w=i时的答案,那么考虑到w=i-1时,如果在每一段的后面加上一个数,就可以得到w=i的情况;但是要注意的是w=i-1时的最后一段是不能通过加上一个数得到w=i的情况的,所以这种情况要去掉,这种情况的数目是很容易计算出来的,因为只需要固定起点为最后一个元素,然后依次向前推看当前区间有多少个不同的元素;然后难点在于如何计算在w=i-1情况下,对每一段加上一个数后增加的情况数;对于这一点我们可以提前预处理出每个数和其最近的相同数的距离来计算(如果没有相同的数,就是和原点的距离),我们统计出各种不同距离的情况数后,就可以利用递推的方法计算出w=i-1时通过在每一段加一个数后增加的情况数了。更具体的见代码:
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=1000010;
long long dp[maxn];
int sum[maxn],pre[maxn],dist[maxn];
int last[maxn],a[maxn];
int main()
{
//freopen("in.txt","r",stdin);
int n;
while(scanf("%d",&n)&&n)
{
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
memset(pre,0,sizeof(pre));
last[1]=1;
pre[a[n]]=1;
int cnt=2;
for(int i=n-1;i>=1;i--)
{
if(pre[a[i]])
last[cnt]=last[cnt-1];
else
{
last[cnt]=last[cnt-1]+1;
pre[a[i]]=1;
}
cnt++;
}
memset(pre,0,sizeof(pre));
memset(dist,0,sizeof(dist));
for(int i=1;i<=n;i++)
{
dist[i-pre[a[i]]]++;
pre[a[i]]=i;
}
sum[n]=dist[n];
for(int i=n-1;i>=1;i--)
{
sum[i]=sum[i+1]+dist[i];
}
dp[1]=n;
for(int i=2;i<=n;i++)
dp[i]=dp[i-1]-last[i-1]+sum[i];
int m;
scanf("%d",&m);
while(m--)
{
int x;
scanf("%d",&x);
printf("%lld\n",dp[x]);
}
}
return 0;
}