BZOJ3019 : [Balkan2012]handsome

首先预处理出$f[i][j][k]$表示长度为$i$的序列，第一个位置是$j$，最后一个位置是$k$时合法的方案数。

从后往前枚举LCP以及那个位置应该改成什么。

用线段树维护区间内最左最右的已经确定的位置，以及区间内的合法方案数。

合并的时候只需要将左右儿子的答案乘起来，然后再乘以左儿子最右到右儿子最左这一段区间的方案数即可。

时间复杂度$O(n\log n)$。

 

#include<cstdio>
const int N=400010,M=1050000,P=1000000007;
int n,m,i,j,k,x,a[N],g[3][3],f[N][3][3],pos[N],v[N],l[M],r[M],val[M],ans=1;char ch[9],s[N];
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
inline void up(int&x,int y){x+=y;if(x>=P)x-=P;}
void build(int x,int a,int b){
  l[x]=a,r[x]=b,val[x]=1;
  if(a==b){pos[a]=x;return;}
  int mid=(a+b)>>1;
  build(x<<1,a,mid),build(x<<1|1,mid+1,b);
}
inline bool change(int x,int p){
  if(~p){
    if(x>1&&~v[x-1])if(g[v[x-1]][p])return 0;
    if(x<n&&~v[x+1])if(g[p][v[x+1]])return 0;
  }
  v[x]=p,x=pos[x];
  if(p<0)l[x]=r[x]=0;
  for(x>>=1;x;x>>=1){
    l[x]=l[x<<1]?l[x<<1]:l[x<<1|1];
    r[x]=r[x<<1|1]?r[x<<1|1]:r[x<<1];
    val[x]=1LL*val[x<<1]*val[x<<1|1]%P;
    if(r[x<<1]&&l[x<<1|1])val[x]=1LL*val[x]*f[l[x<<1|1]-r[x<<1]+1][v[r[x<<1]]][v[l[x<<1|1]]]%P;
  }
  return 1;
}
inline int ask(){
  int ret=val[1],t,i;
  if(l[1]>1){
    for(t=i=0;i<3;i++)up(t,f[l[1]][i][v[l[1]]]);
    ret=1LL*ret*t%P;
  }
  if(r[1]<n){
    for(t=i=0;i<3;i++)up(t,f[n-r[1]+1][v[r[1]]][i]);
    ret=1LL*ret*t%P;
  }
  return ret;
}
int main(){
  read(n);
  for(i=1;i<=n;i++)read(a[i]);
  read(m);
  while(m--)scanf("%s",ch),g[ch[0]-'1'][ch[1]-'1']=1;
  scanf("%s",s+1);
  for(i=1;i<=n;i++)s[i]-='1',v[i]=s[i];
  for(i=0;i<3;i++)f[1][i][i]=1;
  for(i=1;i<n;i++)for(j=0;j<3;j++)for(k=0;k<3;k++)if(f[i][j][k])for(x=0;x<3;x++)if(!g[k][x])up(f[i+1][j][x],f[i][j][k]);
  build(1,1,n);
  for(i=n;i;change(a[i--],-1))for(j=0;j<s[a[i]];j++)if(change(a[i],j))up(ans,ask());
  return printf("%d",ans),0;
}