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
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define inf 100000005
#define M 40
#define N 210005
#define maxn 300005
#define eps 1e-10
#define zero(a) fabs(a) #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 }
int len,c[N];
int slove()
{
mem(c,0);
for(int i=0;i for(int i=1;i for(int i=0;i root->cnt=1;
root->sum=0;
int ans=0;
for(int i=0;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 }
return 0;
}

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define inf 100000005
#define M 40
#define N 210005
#define maxn 300005
#define eps 1e-10
#define zero(a) fabs(a) #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->p

HDU 4436 str2int（12年天津，SAM） (一)