hdu 4351 Digital root(线段树区间合并)

来源：互联网发布：centos jdk yum 编辑：程序博客网时间：2024/06/10 12:24

题意：给出n个数，m组询问，每次询问区间[L,R]及其子区间的Digital root，并输出最大的5个。

思路：感觉好久没写线段树了，今天开心的写了一发~ 用线段树维护这几个值:整个区间的值sum[](0~9),msk[]当前区间能获得的数的状态(状压表示)，pre[]当前区间的所有前缀区间能得到的数的状态，suff[]当前区间的所有后缀区间能得到的数的状态，剩下就是组合组合，合并的时候枚举下能获得的新的数就行了。。。开始合并直接算的，虽然过了，但是速度有些慢，看了网上的代码，发现是预处理了一发，改了改，果然快了不少……

代码：

#include<iostream>#include<cstdio>#include<cstring>#include<string>#include<algorithm>#include<map>#include<queue>#include<stack>#include<cmath>#include<vector>#define inf 0x3f3f3f3f#define Inf 0x3FFFFFFFFFFFFFFFLL#define eps 1e-9#define pi acos(-1.0)using namespace std;typedef long long ll;const int maxn=100000+10;const int mod=9;int msk[maxn<<2],sum[maxn<<2],pre[maxn<<2],suff[maxn<<2];int num[maxn],val[1024][1024];void Init(){    int z;    for(int x=0;x<(1<<10);++x)        for(int y=0;y<(1<<10);++y)        {            val[x][y]=0;            for(int i=0;i<=mod;++i)                if(x&(1<<i))                for(int j=0;j<=mod;++j)                    if(y&(1<<j))                    {                        z=i+j;                        if(z>=mod)                        {                            z-=mod;                            if(z==0) z=mod;                        }                        val[x][y]|=(1<<z);                    }        }}inline int cal(int x,int y) {return val[x][y];}inline void PushUp(int rt){    sum[rt]=sum[rt<<1]+sum[rt<<1|1];    if(sum[rt]>=mod)    {        sum[rt]-=mod;        if(sum[rt]==0) sum[rt]=mod;    }    msk[rt]=msk[rt<<1]|msk[rt<<1|1];    msk[rt]|=cal(suff[rt<<1],pre[rt<<1|1]);    pre[rt]=pre[rt<<1]|cal(1<<sum[rt<<1],pre[rt<<1|1]);    suff[rt]=suff[rt<<1|1]|cal(1<<sum[rt<<1|1],suff[rt<<1]);}void build(int l,int r,int rt){    if(l==r)    {        if(num[l])        {            num[l]%=mod;            if(num[l]==0) num[l]=mod;        }        sum[rt]=num[l];        msk[rt]=pre[rt]=suff[rt]=1<<sum[rt];        return ;    }    int m=(l+r)>>1;    build(l,m,rt<<1);    build(m+1,r,rt<<1|1);    PushUp(rt);}int Query(int L,int R,int l,int r,int rt,int &p,int &f,int &s){    if(l>=L&&r<=R)    {        p=pre[rt];        f=suff[rt];        s=sum[rt];        return msk[rt];    }    int m=(l+r)>>1;    if(m>=R) return Query(L,R,l,m,rt<<1,p,f,s);    else if(m<L) return Query(L,R,m+1,r,rt<<1|1,p,f,s);    else    {        int p1,p2,f1,f2,s1,s2,m1,m2;        m1=Query(L,R,l,m,rt<<1,p1,f1,s1);        m2=Query(L,R,m+1,r,rt<<1|1,p2,f2,s2);        s=s1+s2;        if(s>=mod)        {            s-=mod;            if(s==0) s=mod;        }        p=p1|cal(1<<s1,p2);        f=f2|cal(1<<s2,f1);        return (m1|m2)|cal(f1,p2);    }}int main(){    //freopen("in.txt","r",stdin);    //freopen("out.txt","w",stdout);    int t,tcase=0;    Init();    scanf("%d",&t);    while(t--)    {        if(tcase) printf("\n");        tcase++;        int n,m;        scanf("%d",&n);        for(int i=1;i<=n;++i)            scanf("%d",&num[i]);        build(1,n,1);        scanf("%d",&m);        printf("Case #%d:\n",tcase);        int L,R,p,f,s,res,tot;        while(m--)        {            scanf("%d%d",&L,&R);            res=Query(L,R,1,n,1,p,f,s);            tot=5;            for(int i=mod;i>=0&&tot;--i)                if(res&(1<<i))                {                    if(tot<5) printf(" ");                    printf("%d",i);                    tot--;                }            while(tot)            {                if(tot<5) printf(" ");                printf("-1");                tot--;            }            printf("\n");        }    }    return 0;}

0 0