bzoj3473字符串&bzoj3277串

题意：给定n个字符串，询问每个字符串有多少子串（不包括空串）是所有n个字符串中至少k个字符串的子串.注意本质相同的子串多次出现算多次,如1 1 aaa这组数据答案为6,贡献1WA.代码里有些部分是为了处理子串本质不同,可能没删干净.

因为字符串的总长不超过10^5,那么后缀的个数也不超过10^5。一个长为x的后缀可以产生x个子串，其中在至少k个串中出现的子串一定是长度从1 开始递增的连续几个.那么对每个后缀可以二分找出最多能够产生多么长的在至少k个串中出现的子串.把所有串连起来求后缀数组，二分一个长度mid之后可以找出有哪些后缀和当前后缀的lcp>=mid,那么长度为mid的子串就可以在这些位置出现(即这些后缀所在的字符串包含了这个长度为mid的子串)。这些后缀一定是排名连续的一段区间，预处理每个后缀属于哪个字符串，判断这段区间中出现的不同的归属种数是否大于等于k，那么主席树（参考HH的项链）就可以做了。时间复杂度O(nlog^2n)=O(跑得出)。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=;
int str[maxn];int tot=;
int tmp[][maxn],sum[maxn],key[maxn];int sa[maxn],rank[maxn],height[maxn];
int st[maxn][],qlog[maxn];
void getsa(int n,int m){
  int i,j,k,p,*rk=tmp[],*res=tmp[];
  for(i=;i<m;++i)sum[i]=;
  for(i=;i<n;++i)sum[rk[i]=str[i]]++;
  for(i=;i<m;++i)sum[i]+=sum[i-];
  for(i=n-;i>=;--i){
    sa[--sum[rk[i]]]=i;
  }
  for(j=,p=;p<n;j<<=,m=p){
    for(i=;i<m;++i)sum[i]=;
    for(p=,i=n-j;i<n;++i)res[p++]=i;
    for(i=;i<n;++i)if(sa[i]>=j)res[p++]=sa[i]-j;
    for(i=;i<n;++i)sum[key[i]=rk[res[i]]]++;
    for(i=;i<m;++i)sum[i]+=sum[i-];
    for(i=n-;i>=;--i)sa[--sum[key[i]]]=res[i];
    for(res[sa[]]=,p=,i=;i<n;++i)
      res[sa[i]]=(rk[sa[i]]==rk[sa[i-]]&&rk[sa[i]+j]==rk[sa[i-]+j])?p-:p++;
    swap(res,rk);
  }
  for(i=;i<n;++i)rank[sa[i]]=i;
  for(i=,k=;i<n-;height[rank[i++]]=k)
    for(k?k--:,j=sa[rank[i]-];str[j+k]==str[i+k];++k);
  for(int i=;i<n;++i)st[i][]=height[i];
  for(int j=;(<<j)<n;++j){
    for(int i=;i<n;++i){
      st[i][j]=st[i][j-];
      if(i+(<<j-)<n&&st[i+(<<j-)][j-]<st[i][j])st[i][j]=st[i+(<<j-)][j-];
    }
  }
  for(int j=;(<<j)<n;++j)qlog[<<j]=j;
  for(int i=;i<n;++i)if(!qlog[i])qlog[i]=qlog[i-];
}
int lcp(int a,int b){
  if(a==b)return 0x7f7f7f7f;
  a=rank[a];b=rank[b];
  if(a>b)swap(a,b);
  a++;int j=qlog[b-a+];
  return min(st[a][j],st[b-(<<j)+][j]);
}
char buf[maxn];
int len[maxn],sumlen[maxn];
int belong[maxn];
struct node{
  int sum;node* ch[];
  node(){}
  node(int x){sum=x;ch[]=ch[]=;}
}t[maxn*];int szoftree=;
node* newnode(int x){
  t[++szoftree]=node(x);return t+szoftree;
}
void Insert(node* rt0,node* &rt,int l,int r,int k){
  rt=newnode(rt0->sum+);
  if(l==r)return;
  int mid=(l+r)>>;
  if(k<=mid){Insert(rt0->ch[],rt->ch[],l,mid,k);rt->ch[]=rt0->ch[];}
  else {Insert(rt0->ch[],rt->ch[],mid+,r,k);rt->ch[]=rt0->ch[];}
}
int query(node* rt0,node* rt1,int l,int r,int ql,int qr){
  if(ql>qr)return ;
  if(ql<=l&&r<=qr)return rt1->sum-rt0->sum;
  int mid=(l+r)>>,ans=;
  if(ql<=mid)ans+=query(rt0->ch[],rt1->ch[],l,mid,ql,qr);
  if(qr>mid) ans+=query(rt0->ch[],rt1->ch[],mid+,r,ql,qr);
  return ans;
}
int last[maxn];
node* root[maxn];
typedef long long ll;
ll ans[maxn];
int lastsuf[maxn];
int pos;
int binary2(int l,int r,int x){
  while(l<=r){
    int mid=(l+r)>>;
    if(lcp(pos,sa[mid])>=x)r=mid-;
    else l=mid+;
  }
  return r+;
}
int binary3(int l,int r,int x){
  while(l<=r){
    int mid=(l+r)>>;
    if(lcp(pos,sa[mid])>=x)l=mid+;
    else r=mid-;
  }
  return l-;
}
int n,k;
int binary1(int suf,int l,int r){
  pos=sa[suf];
  while(l<=r){
    int mid=(l+r)>>;
    int left=binary2(,suf,mid),right=binary3(suf,tot,mid);//printf("%d %d %d\n",suf,left,right);
    if(query(root[left-],root[right],,tot,,left)>=k)l=mid+;
    else r=mid-;
  }
  return l-;
}
bool vis[maxn];
int main(){
  scanf("%d%d",&n,&k);
  for(int i=;i<=n;++i){
    scanf("%s",buf);len[i]=strlen(buf);
    for(int j=;j<len[i];++j){
      str[tot++]=buf[j];
    }
    str[tot++]=+i;sumlen[i]=tot;
  }
  str[tot++]=;
  getsa(tot,+n+);
  for(int i=;i<=n;++i){
    for(int j=sumlen[i-];j<sumlen[i]-;++j)belong[rank[j]]=i;
  }
  root[]=t+;root[]->ch[]=root[]->ch[]=t+;root[]->sum=;
  for(int i=;i<tot;++i){
    root[i]=root[i-];
    if(belong[i]){
      Insert(root[i],root[i],,tot,last[belong[i]]+);
      last[belong[i]]=i;
    }
  }//printf("tot==%d\n",tot);
  for(int i=;i<tot;++i){
    if(belong[i]){//printf("%d\n",sa[i]);
      int lo=,hi=sumlen[belong[i]]-sa[i]-;//长度上下界
      // if(vis[belong[i]]){//printf("!");
      //     lo=max(lo,lcp(lastsuf[belong[i]],sa[i])+1);
      // }
      // lastsuf[belong[i]]=sa[i];vis[belong[i]]=true;
      hi=binary1(i,lo,hi);//printf("%d %d\n",lo,hi);
      ans[belong[i]]+=(hi-lo+);
    }
  }
  for(int i=;i<=n;++i)printf("%lld ",ans[i]);
  return ;
}