hdu 4455 Substrings

题意：给定一个整数串，有Q组询问，问这个串中长度为W的子串中不同的数字之和为多少。

可以DP做，也可以用树状数组做。假设当前的数是a[i]，如果它前面也出现了a[i]'，那么a[i]'不计，计入的是a[i]，如果a[i]后面出现了a[i]''，那么a[i]不计，计入的是a[i]''。定义a[i]前面的空位是bef，后面的空格是aft，则W在1-min(bef,aft)这段区间内的，a[i]的是要计算W次的，在min(bef,aft+1)-max(bef,aft)这段区间内，a[i]是要计算min(bef,aft)次的，在max(bef,aft)+1-bef+aft-1这段范围内，a[i]是要计算-W+bef+aft次的。明显看出这是一个一次函数。用两棵树状数组维护它们的系数就可以了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <string>
#include <set>
#include <map>
#include <cmath>
#include <algorithm>
#include <queue>
using namespace std;

typedef long long LL;
const int N=1e6+15;

int pre[N],a[N],mx;
int n,m;

struct node
{
	int valu,pos;
	node(){}
	node(int valu,int pos) :
		valu(valu),pos(pos) {}
	bool operator <(const node &b)const
	{
		return valu<b.valu||(valu==b.valu&&pos<b.pos);
	}
}data[N];
struct BIT
{
	LL T[N];

	int lowbit(int x){return x&(-x);}
	void add(int st,int ed,int valu)
	{
		if(st>ed) return ;
		for(int i=st;i<=n;i+=lowbit(i)) T[i]+=valu;
		for(int i=ed+1;i<=n;i+=lowbit(i)) T[i]-=valu;
	}
	LL query(int pos)
	{
		LL sum=0;
		for(int i=pos;i>0;i-=lowbit(i)) sum+=T[i];
		return sum;
	}
}K,B;
int main()
{
//	freopen("C.in","r",stdin);
//	freopen("C.out","w",stdout);

	while(scanf("%d",&n)!=EOF)
	{
		if(n==0) break;

		mx=0;
		for(int i=1;i<=n;i++) 
		{
			K.T[i]=B.T[i]=0;
			scanf("%d",&a[i]);
			mx=max(mx,a[i]);
		}

		for(int i=0;i<=mx;i++) pre[i]=-1;
		
		for(int i=n;i>=1;i--) 
		{
			int bef=i,aft=n-i+1;

			if(pre[ a[i] ]!=-1)
			{
				aft=min(aft,pre[a[i]]-i);
			}
			pre[ a[i] ]=i;

			K.add(1,min(bef,aft),1);

			int st=min(bef,aft)+1;
			int ed=max(bef,aft);

			if(st<=ed) B.add(st,ed,min(bef,aft));

			if(ed+1<=bef+aft-1)
			{
				K.add(ed+1,bef+aft-1,-1);
				B.add(ed+1,bef+aft-1,bef+aft);
			}
		}
		
		scanf("%d",&m);
		for(int i=0;i<m;i++) 
		{
			LL tmp,ans=0; scanf("%I64d",&tmp);
			ans+=tmp*K.query(tmp);
			ans+=B.query(tmp);
			printf("%I64d\n",ans);
		}
	}
	return 0;
}