ZOJ 2900 Icecream(线段树)

题意：给你一个长度为n的串，从中选出字串，使得长度至少为k，且相邻的数值差的绝对值小于等于p，求这样的字串的个数mod m的值

地址：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=2900

分析：我们可以马上想到，以前i个数，长度为j的字串个数为f[ i ][ j ]，那么有f[ [i ][ j ]=sum{ f[ i' ][ j-1 ] } 1<=i'<=i，但是这样的转移方程是n^3，应该会超时。。。

这种情况下，我们应该看到题目给的提示，这些数值不大于100，从这里下手，便能找到这题的突破口。

由于长度j只与长度j-1的子串有关，所以我们可以枚举串的长度，对于长度为j的第i个元素值f[ i ][ j ]，我们在1~i-1中查找值为a[i]-p~a[i]+p的f[ i' ][ j ]，由于值只有100，所以直接维护一棵线段树，下标为数列的值，保存长度为j-1且末尾为 i‘ 的总数。

额，好久没做题，这题居然用了一个下午，关键连题解也写得好挫

还有一种方法是DP，f[ i ][ j ][ k ]表示前i个元素，长度为j，末尾的值为k的数量，具体方程自己想吧。。。

代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>

#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1

int sum[1111<<2];
int a[2013],f[2][2013];
int i,j,t,g,n,k,p,m,top,ans;

void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    if(sum[rt]>=m)sum[rt]%=m;
}

void build(int l, int r, int rt)
{
    sum[rt]=0;
    if(l==r)return;
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}

void update(int x, int val, int l, int r, int rt)
{
    if(l>=r)
    {
        sum[rt]+=val;
        if(sum[rt]>=m)sum[rt]%=m;
        return;
    }
    int mid=(l+r)>>1;
    if(x<=mid)update(x,val,lson);
    else update(x,val,rson);
    pushup(rt);
}

int query(int L, int R, int l, int r, int rt)
{
    if(L<=l&&R>=r)return sum[rt];
    int mid=(l+r)>>1,ret=0;
    if(L<=mid)ret=query(L,R,lson);
    if(R>mid)ret+=query(L,R,rson);
    if(ret>=m)ret%=m;
    return ret;
}
int main()
{
    while(~scanf("%d%d%d%d",&n,&k,&p,&m))
    {
        ans=0;
        for(top=i=1;i<=n;++i)
        {
            scanf("%d",&a[i]);
            if(a[i]>top)top=a[i];
            f[0][i]=1;
            if(k<=1)++ans;
        }
        if(ans>=m)ans%=m;
        for(g=0,t=2;t<=n;++t,g^=1)
        {
            build(0,top,1);
            for(i=t-1;i<n;++i)
            {
                update(a[i],f[g][i],0,top,1);
                f[g^1][i+1]=query(a[i+1]-p,a[i+1]+p,0,top,1);
                if(t>=k)
                {
                    ans+=f[g^1][i+1];
                    if(ans>=m)ans%=m;
                }
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}