码迷,mamicode.com
首页 > 其他好文 > 详细

hihocoder1457

时间:2018-03-29 22:42:44      阅读:249      评论:0      收藏:0      [点我收藏+]

标签:closed   unsigned   cal   技术分享   pen   false   cli   gcc   fast   

http://hihocoder.com/problemset/problem/1457

找不重复子串的和

topo序搞一搞,用父亲更新儿子节点的val,记得乘上节点数

技术分享图片
//#pragma comment(linker, "/stack:200000000")
//#pragma GCC optimize("Ofast,no-stack-protector")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
//#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define pi acos(-1.0)
#define ll long long
#define mod (1000000007)
#define C 0.5772156649
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define pil pair<int,ll>
#define pii pair<int,int>
#define ull unsigned long long
#define base 1000000000000000000
#define fio ios::sync_with_stdio(false);cin.tie(0)

using namespace std;

const double g=10.0,eps=1e-12;
const int N=2000000+10,maxn=1200000+10,inf=0x3f3f3f3f,INF=0x3f3f3f3f3f3f3f3f;

inline void sub(ll &x){if(x>=mod)x-=mod;}
char s[N];
int c[N],a[N];
ll val[N],sum[N];
struct SAM{
    int last,cnt;
    int ch[N<<1][26],fa[N<<1],l[N<<1];
    int mx[N<<1],tmp[N<<1];
    void ins(int c)
    {
        int p=last,np=++cnt;last=np;l[np]=l[p]+1;
        for(;p&&!ch[p][c];p=fa[p])ch[p][c]=np;
        if(!p)fa[np]=1;
        else
        {
            int q=ch[p][c];
            if(l[p]+1==l[q])fa[np]=q;
            else
            {
                int nq=++cnt;l[nq]=l[p]+1;
                memcpy(ch[nq],ch[q],sizeof(ch[q]));
                fa[nq]=fa[q];fa[q]=fa[np]=nq;
                sum[nq]=1ll*c;
                for(;ch[p][c]==q;p=fa[p])ch[p][c]=nq;
            }
        }
        sum[np]=1ll*c;
    }
    SAM()
    {
        cnt=1;
    }
    void build(){
        int len=strlen(s+1);
        last=1;
        for(int i=1;i<=len;i++)ins(s[i]-0);
//        topo();
//        for(int i=1;i<=cnt;i++)mx[i]=l[i],tmp[i]=0;
    }
    void topo(){
        for(int i=1;i<=cnt;i++)c[l[i]]++;
        for(int i=1;i<=cnt;i++)c[i]+=c[i-1];
        for(int i=1;i<=cnt;i++)a[c[l[i]]--]=i;
    }
    void debug()
    {
        puts("");
        for(int i=1;i<=cnt;i++)
        {
            printf("%d ",i);
            for(int j=0;j<26;j++)
                printf("%d ",ch[i][j]);
            printf("%d %d %lld %lld\n",fa[i],l[i],val[i],sum[i]);
        }
    }
    void faupdate()
    {
        for(int i=cnt;i;i--)
            tmp[fa[a[i]]]=max(tmp[fa[a[i]]],tmp[a[i]]);
    }
    void maupdate()
    {
        for(int i=1;i<=cnt;i++)
            mx[i]=min(mx[i],tmp[i]),tmp[i]=0;
    }
    void match()
    {
        int len=strlen(s+1),now=1,res=0;
        for(int i=1;i<=len;i++)
        {
            if(ch[now][s[i]-a])
            {
                now=ch[now][s[i]-a];
                res++;
            }
            else
            {
                while(now&&!ch[now][s[i]-a])now=fa[now];
                if(!now)now=1,res=0;
                else
                {
                    res=l[now]+1;
                    now=ch[now][s[i]-a];
                }
            }
            tmp[now]=max(tmp[now],res);
        }
        faupdate();
        maupdate();
    }
    int maxlen()
    {
        int ans=0;
        for(int i=1;i<=cnt;i++)ans=max(ans,mx[i]);
        return ans;
    }
    void cal()
    {
        topo();
        for(int i=1;i<=cnt;i++)
        {
            int p=a[i];
            val[p] =  val[p] + 1ll*sum[p] * (l[p] - l[fa[p]])%mod;
            sub(val[p]);
            for(int j=0;j<10;j++)
            {
                if(ch[p][j])
                {
                    val[ch[p][j]]=val[ch[p][j]]+val[p]*10ll%mod;
                    sub(val[ch[p][j]]);
                }
            }
         //   cout << val[p] << endl;
        }
       // debug();
        ll ans=0;
        for(int i=cnt;i;i--)
        {
            ans=ans+val[i];
            sub(ans);
        }
        printf("%lld\n",ans);
    }
}sam;
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s+1);
        sam.build();
    }
    sam.cal();
    return 0;
}
/********************
1
101
********************/
版本1
技术分享图片
//#pragma comment(linker, "/stack:200000000")
//#pragma GCC optimize("Ofast,no-stack-protector")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
//#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define pi acos(-1.0)
#define ll long long
#define mod (1000000007)
#define C 0.5772156649
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define pil pair<int,ll>
#define pii pair<int,int>
#define ull unsigned long long
#define base 1000000000000000000
#define fio ios::sync_with_stdio(false);cin.tie(0)

using namespace std;

const double g=10.0,eps=1e-12;
const int N=2000000+10,maxn=1200000+10,inf=0x3f3f3f3f,INF=0x3f3f3f3f3f3f3f3f;

inline void sub(ll &x){if(x>=mod)x-=mod;}
char s[N];
int c[N],a[N];
ll val[N],sum[N],sz[N];
struct SAM{
    int last,cnt;
    int ch[N<<1][26],fa[N<<1],l[N<<1];
    int mx[N<<1],tmp[N<<1];
    void ins(int c)
    {
        int p=last,np=++cnt;last=np;l[np]=l[p]+1;
        for(;p&&!ch[p][c];p=fa[p])ch[p][c]=np;
        if(!p)fa[np]=1;
        else
        {
            int q=ch[p][c];
            if(l[p]+1==l[q])fa[np]=q;
            else
            {
                int nq=++cnt;l[nq]=l[p]+1;
                memcpy(ch[nq],ch[q],sizeof(ch[q]));
                fa[nq]=fa[q];fa[q]=fa[np]=nq;
                for(;ch[p][c]==q;p=fa[p])ch[p][c]=nq;
            }
        }
        sum[np]=1ll*c;
    }
    SAM()
    {
        cnt=1;
    }
    void build(){
        int len=strlen(s+1);
        last=1;
        for(int i=1;i<=len;i++)ins(s[i]-0);
//        topo();
//        for(int i=1;i<=cnt;i++)mx[i]=l[i],tmp[i]=0;
    }
    void topo(){
        for(int i=1;i<=cnt;i++)c[l[i]]++;
        for(int i=1;i<=cnt;i++)c[i]+=c[i-1];
        for(int i=1;i<=cnt;i++)a[c[l[i]]--]=i;
    }
    void debug()
    {
        puts("");
        for(int i=1;i<=cnt;i++)
        {
            printf("%d ",i);
            for(int j=0;j<26;j++)
                printf("%d ",ch[i][j]);
            printf("%d %d %lld %lld\n",fa[i],l[i],val[i],sum[i]);
        }
    }
    void faupdate()
    {
        for(int i=cnt;i;i--)
            tmp[fa[a[i]]]=max(tmp[fa[a[i]]],tmp[a[i]]);
    }
    void maupdate()
    {
        for(int i=1;i<=cnt;i++)
            mx[i]=min(mx[i],tmp[i]),tmp[i]=0;
    }
    void match()
    {
        int len=strlen(s+1),now=1,res=0;
        for(int i=1;i<=len;i++)
        {
            if(ch[now][s[i]-a])
            {
                now=ch[now][s[i]-a];
                res++;
            }
            else
            {
                while(now&&!ch[now][s[i]-a])now=fa[now];
                if(!now)now=1,res=0;
                else
                {
                    res=l[now]+1;
                    now=ch[now][s[i]-a];
                }
            }
            tmp[now]=max(tmp[now],res);
        }
        faupdate();
        maupdate();
    }
    int maxlen()
    {
        int ans=0;
        for(int i=1;i<=cnt;i++)ans=max(ans,mx[i]);
        return ans;
    }
    void cal()
    {
        topo();
        sz[1]=1;
        for(int i=1;i<=cnt;i++)
        {
            int p=a[i];
//            val[p] =  val[p] + 1ll*sum[p] * (l[p]-l[fa[p]])%mod;
//            sub(val[p]);
            for(int j=0;j<10;j++)
            {
                if(ch[p][j])
                {
                    val[ch[p][j]]=val[ch[p][j]]+val[p]*10ll%mod+1ll*j*sz[p]%mod;
                    sub(val[ch[p][j]]);
                    sz[ch[p][j]]+=sz[p];
                }
            }
         //   cout << val[p] << endl;
        }
       // debug();
        ll ans=0;
        for(int i=cnt;i;i--)
        {
            ans=ans+val[i];
            sub(ans);
        }
        printf("%lld\n",ans);
    }
}sam;
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s+1);
        sam.build();
//        if(i!=n)sam.ins(10);
    }
    sam.cal();
    return 0;
}
/********************
1
987654321123456789
********************/
版本2

 

hihocoder1457

标签:closed   unsigned   cal   技术分享   pen   false   cli   gcc   fast   

原文地址:https://www.cnblogs.com/acjiumeng/p/8672595.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!