第一个点超时,怎么优化(求答谢谢)(在线等)

回复帖子

@寻旧 2019-05-07 17:15 回复
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
int T,n,ch[N][30],bo[N],nxt[N];
int id,ans;
queue<int>q;
inline void add(char *s)
{
    int u=1,len=strlen(s+1);
    for(int i=1;i<=len;++i)
    {
        int c=s[i]-'a'+1;
        if(!ch[u][c]) ch[u][c]=++id;
        u=ch[u][c];
    }
    bo[u]++;
}
inline void bfs()
{
    q.push(1);nxt[1]=0;
    while(q.size())
    {
        int u=q.front();q.pop();
        for(int i=1;i<=26;++i)
        {
            if(!ch[u][i]) ch[u][i]=ch[nxt[u]][i];
            else
            {
                q.push(ch[u][i]);
                int v=nxt[u];
                nxt[ch[u][i]]=ch[v][i];
            }
        }
    }
}
inline void find(char *s)
{
    int u=1,len=strlen(s+1);
    for(int i=1;i<=len;++i)
    {
        int c=s[i]-'a'+1;
        int k=ch[u][c];
        while(k>1)
        {
            ans+=bo[k];
            bo[k]=0;
            k=nxt[k];
        }
        u=ch[u][c];
    }
}
int main()
{
    char s[N];  
    ans=0;id=1;
    for(int i=1;i<=26;++i) ch[0][i]=1;
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
    {
        scanf("%s",s+1);
        add(s);
    }
    bfs();
    scanf("%s",s+1);
    find(s);
    printf("%d\n",ans);
    return 0;
}

@寻旧 find 函数中打标记,已经扫到过的节点没必要扫多次。具体见我代码。

// luogu-judger-enable-o2
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
int n,ch[5010000][26];
int tot,f[5010000],nxt[5010000];
char s[1010000];
inline void add(){
    int len=strlen(s+1),now=1;
    for(int i=1;i<=len;i++){
        int u=s[i]-'a';
        if(!ch[now][u]){
            ch[now][u]=++tot;
        }
        now=ch[now][u];
    }
    f[now]++;
}
queue<int>q;
inline void get_nxt(){
    for(int i=0;i<26;i++)ch[0][i]=1;
    nxt[1]=0;
    q.push(1);
    while(!q.empty()){
        int x=q.front();
        q.pop();
        for(int i=0;i<26;i++){
            int u=ch[x][i];
            if(!u){
                ch[x][i]=ch[nxt[x]][i];
            }
            else{
                int p=nxt[x];
                q.push(u);
                nxt[u]=ch[p][i];
            }
        }
    }
}
inline int find(){
    int len=strlen(s+1);
    int i=1,now=1,q,w,ans=0;
    while(i<=len){
        q=s[i]-'a';
        w=ch[now][q];
        while(w>1){
            if(f[w]==-1)break;
            ans+=f[w];
            f[w]=-1;
            w=nxt[w];
        }
        now=ch[now][q];
        i++;
    }
    return ans;
}
int main(){
    scanf("%d",&n);
    tot=1;
    for(int i=1;i<=n;i++){
        scanf("%s",s+1);
        add();
    }
    get_nxt();
    scanf("%s",s+1);
    printf("%d\n",find());
    return 0;
}
反馈
如果你认为某个帖子有问题,欢迎向洛谷反馈,以帮助更多的同学。



请具体说明理由,以增加反馈的可信度。