[回文串 线段树] Codeforces Gym100032 ICL Cup 2012 K. Subpalindromes

题意是求一个区间里回文串的个数(出现位置不同的回文串算不同)

用马拉车算出以每个点为中心的最长回文串长度

考虑点 ii点到以它为中点的最长的回文串的端点长度为 x

那么它对一个询问的贡献是 min{iL,Ri,x}

把询问的区间分成两部分 [L,mid][mid,R]

那么若 i[L,mid] 则贡献为 min{iL,x},这个东西就相当于求 [L,mid][ix,x] 相交部分的长度,用线段树区间覆盖区间查询搞搞就行了

细节贼多

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#define fi first
#define se second

using namespace std;

typedef pair<int,int> pii;
typedef long long ll;

const int N=500010;

char a[N],c[N];
int n,m,l[N];
ll ans[N];

void HorsePullCar(char *a,int n){
  int pos=0,lst=0;
  for(int i=1;i<=n;i++){
    int j=1;
    if(i<=lst) j=min(lst-i+1,l[pos*2-i]);
    while(a[i+j]==a[i-j]) j++;
    l[i]=j;
    if(i+l[i]-1>lst) lst=i+l[i]-1,pos=i;
  }
}

int tag[N<<2],size[N<<2];
ll tot[N<<2];

vector<pii> add[N];
vector<pii> Ql[N],Qr[N];

inline void Push(int g){
  if(tag[g]){
    tot[g<<1]+=1LL*size[g<<1]*tag[g];
    tot[g<<1|1]+=1LL*size[g<<1|1]*tag[g];
    tag[g<<1]+=tag[g]; tag[g<<1|1]+=tag[g];
    tag[g]=0;
  }
}

inline void Up(int g){
  tot[g]=tot[g<<1]+tot[g<<1|1];
}

void Add(int g,int l,int r,int L,int R){
  if(l==L && R==r) return tot[g]+=size[g],tag[g]++,void();
  int mid=L+R>>1; Push(g);
  if(r<=mid) Add(g<<1,l,r,L,mid);
  else if(l>mid) Add(g<<1|1,l,r,mid+1,R);
  else Add(g<<1,l,mid,L,mid),Add(g<<1|1,mid+1,r,mid+1,R);
  Up(g);
}

ll Query(int g,int l,int r,int L,int R){
  if(l==L && r==R) return tot[g];
  int mid=L+R>>1; Push(g);
  if(r<=mid) return Query(g<<1,l,r,L,mid);
  else if(l>mid) return Query(g<<1|1,l,r,mid+1,R);
  else return Query(g<<1,l,mid,L,mid)+Query(g<<1|1,mid+1,r,mid+1,R);
}

void PutAns(ll x){
  if(x>=10) PutAns(x/10); putchar(x%10+'0');
}

void build(int g,int l,int r){
  size[g]=r-l+1;
  if(l==r) return ;
  int mid=l+r>>1;
  build(g<<1,l,mid); build(g<<1|1,mid+1,r);
}

int main(){
  freopen("1.in","r",stdin);
  freopen("1.out","w",stdout);
  scanf("%s",a+1); n=strlen(a+1);
  int t=0;
  for(int i=1;i<=n;i++)
    c[++t]=a[i],c[++t]='$';
  HorsePullCar(c,t);
  scanf("%d",&m); build(1,1,n);
  for(int i=1;i<=t;i++)
    if(l[i]>1 || (i&1)){
      if(i&1) add[i+1>>1].push_back(pii((i-l[i]+1)/2+1,(i+1)/2)); 
      else add[(i+1)/2+1].push_back(pii((i-l[i]+1)/2+1,(i+1)/2));
    }
  for(int i=1;i<=m;i++){
    int l,r,mid; scanf("%d%d",&l,&r);
    if(l==r){
      ans[i]=1; continue;
    }
    mid=l+r>>1;
    Ql[mid].push_back(pii(l,i));
    Qr[mid+1].push_back(pii(r,i));
  }
  for(int i=1;i<=n;i++){
    for(int j=0;j<add[i].size();j++)
      Add(1,add[i][j].fi,add[i][j].se,1,n);
    add[i].clear();
    for(int j=0;j<Ql[i].size();j++)
      ans[Ql[i][j].se]+=Query(1,Ql[i][j].fi,i,1,n);
  }
  for(int i=1;i<=t;i++)
    if(l[i]>1 || (i&1)){
      if(i&1) add[(i+1)/2].push_back(pii(i+1>>1,(i+l[i])/2));
      else add[(i+1)/2+1].push_back(pii((i+1)/2+1,(i+l[i])/2));
    }
  memset(tot,0,sizeof(tot)); memset(tag,0,sizeof(tag));
  for(int i=n;i;i--){
    for(int j=0;j<add[i].size();j++)
      Add(1,add[i][j].fi,add[i][j].se,1,n);
    for(int j=0;j<Qr[i].size();j++)
      ans[Qr[i][j].se]+=Query(1,i,Qr[i][j].fi,1,n);
  }
  for(int i=1;i<=m;i++) PutAns(ans[i]),putchar('\n');
  return 0;
}
阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭