程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> HDU 4436 str2int(12年天津,SAM)

HDU 4436 str2int(12年天津,SAM)

編輯:C++入門知識

類似SA一樣,在串之間加上特殊字符,這裡用數字10就行了,建立SAM。

然後對SAM進行拓撲一次,從前往後統計。

記錄到達該節點的路徑數量cnt,以前到達該結點的和sum。

則子節點的和則加上,當前節點的和*10+cnt*j,j表示在末尾添加的數字。

拓撲部分,可以用經典的線性,也可以暴力對Len排序。

但是有一些要注意的地方:

1、10是添加的特殊字符,關於10的邊是不轉移的,這比較顯然,只不過是為了不計算重復子串,將所有串拼在一起

2、前導0是要忽略的,也就是從root轉移的時候,不能走0,不然會重復計算


[cpp]
#include<iostream>  
#include<cstdio>  
#include<map>  
#include<cstring>  
#include<cmath>  
#include<vector>  
#include<algorithm>  
#include<set>  
#include<string>  
#include<queue>  
#define inf 100000005  
#define M 40  
#define N 210005  
#define maxn 300005  
#define eps 1e-10  
#define zero(a) fabs(a)<eps  
#define Min(a,b) ((a)<(b)?(a):(b))  
#define Max(a,b) ((a)>(b)?(a):(b))  
#define pb(a) push_back(a)  
#define mp(a,b) make_pair(a,b)  
#define mem(a,b) memset(a,b,sizeof(a))  
#define LL unsigned long long  
#define MOD 2012  
#define lson step<<1  
#define rson step<<1|1  
#define sqr(a) ((double)(a)*(a))  
#define Key_value ch[ch[root][1]][0]  
#define test puts("OK");  
#pragma comment(linker, "/STACK:1024000000,1024000000")  
using namespace std; 
struct SAM 

    SAM *pre,*son[11]; 
    int len,cnt,sum; 
}*root,*tail,que[N],*b[N]; 
int tot; 
char str[N/2]; 
void add(int c,int l) 

    SAM *p=tail,*np=&que[tot++]; 
    np->len=l; 
    while(p&&p->son[c]==NULL) p->son[c]=np,p=p->pre; 
    if(p==NULL) np->pre=root; 
    else 
    { 
        SAM *q=p->son[c]; 
        if(p->len+1==q->len) np->pre=q; 
        else 
        { 
            SAM *nq=&que[tot++]; 
            *nq=*q; 
            nq->len=p->len+1; 
            np->pre=q->pre=nq; 
            while(p&&p->son[c]==q) p->son[c]=nq,p=p->pre; 
        } 
    } 
    tail=np; 

bool cmp(int x,int y) 

    return que[x].len<que[y].len; 

int len,c[N]; 
int slove() 

    mem(c,0); 
    for(int i=0;i<tot;i++) c[que[i].len]++; 
    for(int i=1;i<len;i++) c[i]+=c[i-1]; 
    for(int i=0;i<tot;i++) b[--c[que[i].len]]=&que[i]; 
    root->cnt=1; 
    root->sum=0; 
    int ans=0; 
    for(int i=0;i<tot;i++) 
    { 
        SAM *p=b[i]; 
        for(int j=0;j<10;j++) 
        { 
            if(i==0&&j==0) continue; 
            if(p->son[j]) 
            { 
                SAM *q=p->son[j]; 
                q->cnt=(q->cnt+p->cnt)%MOD; 
                q->sum=(q->sum+p->sum*10+p->cnt*j)%MOD; 
            } 
        } 
        ans=(ans+p->sum)%MOD; 
    } 
    return ans; 

int main() 

    int n; 
    while(scanf("%d",&n)!=EOF) 
    { 
        tot=0; 
        root=tail=&que[tot++]; 
        len=1; 
        while(n--) 
        { 
            scanf("%s",str); 
            for(int i=0;str[i];i++) add(str[i]-'0',len++); 
            add(10,len++); 
        } 
        printf("%d\n",slove()); 
        for(int i=0;i<tot;i++) que[i].sum=que[i].cnt=0,que[i].pre=NULL,mem(que[i].son,NULL); 
    } 
    return 0; 

#include<iostream>
#include<cstdio>
#include<map>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
#include<set>
#include<string>
#include<queue>
#define inf 100000005
#define M 40
#define N 210005
#define maxn 300005
#define eps 1e-10
#define zero(a) fabs(a)<eps
#define Min(a,b) ((a)<(b)?(a):(b))
#define Max(a,b) ((a)>(b)?(a):(b))
#define pb(a) push_back(a)
#define mp(a,b) make_pair(a,b)
#define mem(a,b) memset(a,b,sizeof(a))
#define LL unsigned long long
#define MOD 2012
#define lson step<<1
#define rson step<<1|1
#define sqr(a) ((double)(a)*(a))
#define Key_value ch[ch[root][1]][0]
#define test puts("OK");
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
struct SAM
{
    SAM *pre,*son[11];
    int len,cnt,sum;
}*root,*tail,que[N],*b[N];
int tot;
char str[N/2];
void add(int c,int l)
{
    SAM *p=tail,*np=&que[tot++];
    np->len=l;
    while(p&&p->son[c]==NULL) p->son[c]=np,p=p->pre;
    if(p==NULL) np->pre=root;
    else
    {
        SAM *q=p->son[c];
        if(p->len+1==q->len) np->pre=q;
        else
        {
            SAM *nq=&que[tot++];
            *nq=*q;
            nq->len=p->len+1;
            np->pre=q->pre=nq;
            while(p&&p->son[c]==q) p->son[c]=nq,p=p->pre;
        }
    }
    tail=np;
}
bool cmp(int x,int y)
{
    return que[x].len<que[y].len;
}
int len,c[N];
int slove()
{
    mem(c,0);
    for(int i=0;i<tot;i++) c[que[i].len]++;
    for(int i=1;i<len;i++) c[i]+=c[i-1];
    for(int i=0;i<tot;i++) b[--c[que[i].len]]=&que[i];
    root->cnt=1;
    root->sum=0;
    int ans=0;
    for(int i=0;i<tot;i++)
    {
        SAM *p=b[i];
        for(int j=0;j<10;j++)
        {
            if(i==0&&j==0) continue;
            if(p->son[j])
            {
                SAM *q=p->son[j];
                q->cnt=(q->cnt+p->cnt)%MOD;
                q->sum=(q->sum+p->sum*10+p->cnt*j)%MOD;
            }
        }
        ans=(ans+p->sum)%MOD;
    }
    return ans;
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        tot=0;
        root=tail=&que[tot++];
        len=1;
        while(n--)
        {
            scanf("%s",str);
            for(int i=0;str[i];i++) add(str[i]-'0',len++);
            add(10,len++); www.2cto.com
        }
        printf("%d\n",slove());
        for(int i=0;i<tot;i++) que[i].sum=que[i].cnt=0,que[i].pre=NULL,mem(que[i].son,NULL);
    }
    return 0;
}

 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved