The Preliminary Contest for ICPC Asia Xuzhou 2019 Colorful String(回文树+线段树+状压/bitset)

最新推荐文章于 2024-11-05 03:09:53 发布

a657285080

最新推荐文章于 2024-11-05 03:09:53 发布

阅读量116

点赞数

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/kkkek/p/11484022.html

版权

Colorful String

下午比赛TLE，一直很纳闷为什么线段树+回文树会T，然后晚上发现我线段树写错一行。然后气哭QAQ。

113m赛后过，不会T。

下面代码用的是bitset，也可以直接状压，毕竟才26位。

线段树是记录[l,r]区间的状态，最后返回状态再得到1的数量。

回文树还是那个回文树。

#include<bits/stdc++.h>
using namespace std;
const int maxn=3e5+50;
typedef long long ll;
/*******线段树***********/ 
struct P{
    int l,r;
    bitset<30>bit;
}tree[maxn<<2];
inline void build(int l,int r,int k)
{
    tree[k].l=l;tree[k].r=r;   
    if(tree[k].l==tree[k].r)return; 
    int mid=(tree[k].l+tree[k].r)>>1;
    build(l,mid,k<<1);
    build(mid+1,r,k<<1|1);
}
inline void add(int x,int u,int k)
{
    if(tree[k].l==tree[k].r)
    {
        tree[k].bit.set(u);return;
    }
    int mid=(tree[k].l+tree[k].r)>>1;
    if(x<=mid)add(x,u,k<<1);
    else add(x,u,k<<1|1);
    tree[k].bit=tree[k<<1].bit|tree[k<<1|1].bit;
}
inline bitset<30> find(int l,int r,int k)
{
    if(tree[k].l==l&&tree[k].r==r)return tree[k].bit;
    if(tree[k].l==tree[k].r)
    {
        return tree[k].bit;
    }
    int mid=(tree[k].l+tree[k].r)>>1;
    if(r<=mid)return find(l,r,k<<1);
    if(l>mid)return find(l,r,k<<1|1);
    return find(l,mid,k<<1)|find(mid+1,r,k<<1|1);
}
/***************************************/
char s[maxn];//存储主串 
ll ans1[maxn],cnt[maxn],ans;
int i,j,k,m,n,o,p,q,js,jl,tot,last;
int len[maxn],fail[maxn],ch[maxn][26];
inline int newnode(int x)
{
    tot++;
    len[tot]=x;
    return(tot);
}
inline int getfail(int x,int n)
{
    while(s[n-len[x]-1]!=s[n])x=fail[x];
    return(x);
}
int main()
{
    int u;
    scanf("%s",s+1);
    s[0]=-1;fail[0]=1;last=0;
    len[0]=0;len[1]=-1;tot=1;
    int num=strlen(s+1);
    build(1,num,1);
    for(i=1;i<=num;i++)
    {
        u=s[i]-'a';
        add(i,u,1);
    }
    for(i=1;i<=num;i++)
    {
        s[i]-='a';
        p=getfail(last,i);
        if(ch[p][s[i]]==0)
        {
            q=newnode(len[p]+2);
            fail[q]=ch[getfail(fail[p],i)][s[i]];
            ch[p][s[i]]=q;
            int kkk=min(((2*i-len[q]+1)>>1)+1,i);
            ans1[ch[p][s[i]]]=find(i-len[q]+1,kkk,1).count();
        }
        last=ch[p][s[i]];
        cnt[ch[p][s[i]]]++;
    }
    ans=0;
    for(i=tot;i>1;i--)
    {
        cnt[fail[i]]+=cnt[i];
        ans+=cnt[i]*ans1[i];
    }
    printf("%lld\n",ans);
    return 0;
}

还是好气啊。怪我太着急，一T就慌了。得改。

要好好和队友合作，不能急凸(｀⌒´メ)凸

2019-09-08 00:08:30

然后去好好理解了回文树。

直接回文树可以过这道题了。参考师兄的话：每加一个字符判断该字符在last节点是否出现过，可以不用线段树。

搞不明白，为啥子题解还要dfs呢.....

#include<bits/stdc++.h>
using namespace std;
const int maxn=3e5+50;
typedef long long ll;

char s[maxn];//存储主串 
ll ans1[maxn],cnt[maxn],ans;
int i,j,k,m,n,o,p,q,js,jl,tot,last;
//last节点是当前节点的上一个节点 
int len[maxn],fail[maxn],ch[maxn][26];
//len[i]:节点i的回文串长度
//fail[i]:指向节点i的最长后缀回文串的节点 
int bit[maxn]={0},tt;
inline int newnode(int x)
{
    tot++;
    len[tot]=x;
    return(tot);
}
inline int getfail(int x,int n)
{
    while(s[n-len[x]-1]!=s[n])x=fail[x];
    //设s[i]为字母y 找yAy的A串所在节点 
    return(x);
}
inline int getsize(int x)
{
    int ans=0;
    for(int i=0;i<=26&&(1<<i)<=x;i++)
        if(x&(1<<i))ans++;
    return ans;
}
int main()
{
    int u;
    scanf("%s",s+1);
    s[0]=-1;fail[0]=1;last=0;
    len[0]=0;len[1]=-1;tot=1;
    int num=strlen(s+1);
    for(i=1;i<=num;i++)
    {
        s[i]-='a';
        p=getfail(last,i);
        if(ch[p][s[i]]==0)//p节点没有一条标有s[i]字符的边 
        {
            tt=bit[p];
            tt|=(1<<s[i]);
            q=newnode(len[p]+2);
            bit[q]=tt;
            //新建节点q 节点q长度为p节点长度+2：
            //因为 yAy；A是p节点所代表的回文串 
            fail[q]=ch[getfail(fail[p],i)][s[i]];
            //如果p节点的最长回文后缀有一条标有s[i]字符的边，
            //则这条边所连的节点所代表的的回文串就是q节点的最长回文后缀 
            ch[p][s[i]]=q;
            ans1[ch[p][s[i]]]=getsize(bit[q]);
        }
        last=ch[p][s[i]];
        cnt[ch[p][s[i]]]++;
    }
    ans=0;
    for(i=tot;i>1;i--)
    {
        cnt[fail[i]]+=cnt[i];
        ans+=cnt[i]*ans1[i];
    }
    printf("%lld\n",ans);
    return 0;
}