HDU 5469 Antonidas【点分治+hash】

痛定思痛。 2022-08-08 20:00 148阅读 0赞

首先将串的前缀和后缀分别hash。  
然后用树分治。

对于树分治，每一个分治的root，我们首先要加上经过root的串匹配出的答案，其次要减去*v*→*r**o**o**t*→*v*这样的串（不合法）。  
做了一些剪枝，因为如果*a**n**s*>0，则就是匹配到了。  
如果都没有则继续进行分治。

对于匹配答案，我们记录合法的后缀，然后用前缀去匹配。

// whn6325689
    // Mr.Phoebe
    // http://blog.csdn.net/u013007900
    #include <algorithm>
    #include <iostream>
    #include <iomanip>
    #include <cstring>
    #include <climits>
    #include <complex>
    #include <fstream>
    #include <cassert>
    #include <cstdio>
    #include <bitset>
    #include <vector>
    #include <deque>
    #include <queue>
    #include <stack>
    #include <ctime>
    #include <set>
    #include <map>
    #include <cmath>
    #include <functional>
    #include <numeric>
    #pragma comment(linker, "/STACK:1024000000,1024000000")
    
    
    using namespace std;
    
    #define eps 1e-9
    #define PI acos(-1.0)
    #define INF 0x3f3f3f3f
    #define LLINF 1LL<<62
    #define speed std::ios::sync_with_stdio(false);
    
    typedef long long ll;
    typedef unsigned long long ull;
    typedef long double ld;
    typedef pair<ll, ll> pll;
    typedef complex<ld> point;
    typedef pair<int, int> pii;
    typedef pair<pii, int> piii;
    typedef vector<int> vi;
    
    #define CLR(x,y) memset(x,y,sizeof(x))
    #define CPY(x,y) memcpy(x,y,sizeof(x))
    #define clr(a,x,size) memset(a,x,sizeof(a[0])*(size))
    #define cpy(a,x,size) memcpy(a,x,sizeof(a[0])*(size))
    #define debug(a) cout << #a" = " << (a) << endl;
    #define debugarry(a, n) for (int i = 0; i < (n); i++) { cout << #a"[" << i << "] = " << (a)[i] << endl; }
    
    #define mp(x,y) make_pair(x,y)
    #define pb(x) push_back(x)
    #define lowbit(x) (x&(-x))
    
    #define MID(x,y) (x+((y-x)>>1))
    #define getidx(l,r) (l+r | l!=r)
    #define ls getidx(l,mid)
    #define rs getidx(mid+1,r)
    #define lson l,mid
    #define rson mid+1,r
    
    template<class T>
    inline bool read(T &n)
    {
        T x = 0, tmp = 1;
        char c = getchar();
        while((c < '0' || c > '9') && c != '-' && c != EOF) c = getchar();
        if(c == EOF) return false;
        if(c == '-') c = getchar(), tmp = -1;
        while(c >= '0' && c <= '9') x *= 10, x += (c - '0'),c = getchar();
        n = x*tmp;
        return true;
    }
    template <class T>
    inline void write(T n)
    {
        if(n < 0)
        {
            putchar('-');
            n = -n;
        }
        int len = 0,data[20];
        while(n)
        {
            data[len++] = n%10;
            n /= 10;
        }
        if(!len) data[len++] = 0;
        while(len--) putchar(data[len]+48);
    }
    //-----------------------------------
    
    const int MAXN=10010;
    const ull base=31;
    
    struct Edge
    {
        int to,next;
    } e[MAXN<<1];
    char c[MAXN],str[MAXN];
    int head[MAXN],tot;
    int fa[MAXN],sz[MAXN],num[MAXN];
    int root,tot_size;
    int que[MAXN],n,dist[MAXN],L;
    bool vis[MAXN],used[MAXN];
    ull ldist[MAXN],rdist[MAXN],B[MAXN],Pre[MAXN],Suf[MAXN];
    
    void init()
    {
        for(int i=0; i<=n; i++)
        {
            vis[i]=used[i]=0;
            head[i]=-1;
        }
        tot=0;
    }
    
    void addedge(int u,int v)
    {
        e[tot].to=v;
        e[tot].next=head[u];
        head[u]=tot++;
    }
    
    void get_siz(int u,int fa=-1)
    {
        sz[u]=1;
        int v;
        for(int i=head[u]; ~i; i=e[i].next)
        {
            v=e[i].to;
            if(!vis[v] && v!=fa)
            {
                get_siz(v,u);
                sz[u]+=sz[v];
            }
        }
    }
    
    void get_root(int u,int fa,int &root)
    {
        num[u]=0;
        int v;
        for(int i=head[u]; ~i; i=e[i].next)
        {
            v=e[i].to;
            if(!vis[v] && v!=fa)
            {
                get_root(v,u,root);
                num[u]=max(num[u],sz[v]);
            }
        }
        num[u]=max(num[u],tot_size-sz[u]);
        if(num[u]<num[root]) root=u;
    }
    
    int go(int u)
    {
        int l=0,r=0;
        int ans=0;
        for(que[r++]=u; l<r; l++)
        {
            int now=que[l];
            if(dist[now]<L && rdist[now]==Suf[dist[now]])
                used[dist[now]]++;
            for(int i=head[now]; ~i; i=e[i].next)
            {
                int v=e[i].to;
                if(vis[v]||fa[now]==v)  continue;
                fa[v]=now;
                dist[v]=dist[now]+1;
                ldist[v]=ldist[now]+c[v-1]*B[dist[v]];
                rdist[v]=rdist[now]+c[v-1]*B[dist[v]-1];
                que[r++]=v;
            }
        }
        for(int i=0; i<r; i++)
        {
            int v=que[i];
            if(dist[v]<L && ldist[v]==Pre[dist[v]+1]) ans+=used[L-dist[v]-1];
        }
        for(int i=0; i<r; i++)
            used[dist[que[i]]]=0;
        return ans;
    }
    
    int dfs(int u)
    {
        fa[u]=0;dist[u]=0;
        ldist[u]=c[u-1];rdist[u]=0;
        vis[u]=true;
        int ans=0;
        ans+=go(u);
        for(int i=head[u]; ~i; i=e[i].next)
        {
            int v=e[i].to;
            if(vis[v])  continue;
            ans-=go(v);
        }
        if(ans>0)   return ans;
        for(int i=head[u]; ~i; i=e[i].next)
        {
            int v=e[i].to;
            if(vis[v])  continue;
            get_siz(v);
            tot_size=sz[v];root=0;
            get_root(v,-1,root);
            ans+=dfs(root);
            if(ans>0)   return ans;
        }
        return ans;
    }
    
    int main()
    {
        freopen("data.txt","r",stdin);
        B[0]=1;
        for(int i=1; i<MAXN; i++)B[i]=B[i-1]*base;
        int T,cas=1;
        scanf("%d",&T);
        while(T--)
        {
            read(n);
            init();
            for(int i=1,u,v; i<n; i++)
            {
                read(u);read(v);
                addedge(u,v);
                addedge(v,u);
            }
            scanf("%s %s",c,str);
            Pre[0]=0;Suf[0]=0;
            L=strlen(str);
            for(int i=0; i<L; i++)  Pre[i+1]=Pre[i]*base+str[i];
            reverse(str,str+L);
            for(int i=0; i<L; i++)  Suf[i+1]=Suf[i]*base+str[i];
            get_siz(1);
            tot_size=sz[1];root=0;
            get_root(1,-1,root);
            int ans=dfs(root);
            printf("Case #%d: ",cas++);
            if(ans) puts("Find");
            else puts("Impossible");
        }
        return 0;
    }

我更加推崇的树分治方式。

#include<bits/stdc++.h>
    #define MAXN 10005
    using namespace std;
    template<class T>
    inline bool read(T &n){
        T x = 0, tmp = 1; char c = getchar();
        while((c < '0' || c > '9') && c != '-' && c != EOF) c = getchar();
        if(c == EOF) return false;
        if(c == '-') c = getchar(), tmp = -1;
        while(c >= '0' && c <= '9') x *= 10, x += (c - '0'),c = getchar();
        n = x*tmp;
        return true;
    }
    typedef unsigned long long LL;
    const LL base = 31;
    struct edge{
        int to,next;
    }E[MAXN*2];
    char C[MAXN],S[MAXN];
    int head[MAXN],si,fa[MAXN],sz[MAXN],que[MAXN],N,vis[MAXN],dist[MAXN],used[MAXN],L;
    LL ldist[MAXN],rdist[MAXN],B[MAXN],Pre[MAXN],Suf[MAXN];
    void add_edge(int u,int v)
    {
        E[si].to=v;
        E[si].next=head[u];
        head[u]=si++;
    }
    int get_root(int root)
    {
        int l=0,r=0,res=N+1;fa[root]=0;
        for(que[r++]=root;l<r;l++){
            int now=que[l];
            for(int i=head[now];~i;i=E[i].next){
                int u=E[i].to;
                if(vis[u]||fa[now]==u)continue;
                fa[u]=now;
                que[r++]=u;
            }
        }
        for(;r;r--){
            int now=que[r-1], m=0;
            sz[now]=1;
            for(int i=head[now];~i;i=E[i].next){
                int u=E[i].to;
                if(vis[u]||fa[now]==u)continue;
                sz[now]+=sz[u];
                m=max(m,sz[u]);
            }
            m=max(m,l-sz[now]);
            if(res>m){
                res=m;
                root=now;
            }
        }
        return root;
    }
    int go(int v,int root)
    {
        int l=0,r=0;
        int ans=0;
       // printf("go(%d,%d):",v,root);
        for(que[r++]=v;l<r;l++){
            int now=que[l];
            if(dist[now]<L&&rdist[now]==Suf[dist[now]]){
                used[dist[now]]++;
            }
           // printf("dist[%d]=%d___%llu__%llu\n",now,dist[now],ldist[now],rdist[now]);
            for(int i=head[now];~i;i=E[i].next){
                int u=E[i].to;
                if(vis[u]||fa[now]==u)continue;
                fa[u]=now;
                dist[u]=dist[now]+1;
                ldist[u]=ldist[now]+C[u-1]*B[dist[u]];
                rdist[u]=rdist[now]+C[u-1]*B[dist[u]-1];
                que[r++]=u;
            }
        }
        for(int i=0;i<r;i++){
            int u=que[i];
            if(dist[u]<L&&ldist[u]==Pre[dist[u]+1])ans+=used[L-dist[u]-1];
        }
        for(int i=0;i<r;i++){
            int u=que[i];
            used[dist[u]]=0;
        }
       // printf("___%d\n",ans);
        return ans;
    }
    int solve(int root)
    {
        root=get_root(root);fa[root]=0;
      //  printf("root:%d__\n",root);
        dist[root]=0;ldist[root]=C[root-1];rdist[root]=0;vis[root]=true;
        int ans=0;
        ans+=go(root,root);
        for(int i=head[root];~i;i=E[i].next){
            int u=E[i].to;
            if(vis[u])continue;
            ans-=go(u,root);
        }
      //  printf("_____%d\n",ans);
        if(ans>0)return ans;
        for(int i=head[root];~i;i=E[i].next){
            int u=E[i].to;
            if(vis[u])continue;
            ans+=solve(u);
            if(ans>0)return ans;
        }
        return ans;
    }
    int main()
    {
        freopen("in.txt","r",stdin);
        freopen("out.txt","w",stdout);
        B[0]=1;
        for(int i=1;i<MAXN;i++)B[i]=B[i-1]*base;
        int T;scanf("%d",&T);
        for(int cas=1;cas<=T;cas++){
            read(N);
            for(int i=0;i<=N;i++){
                vis[i]=used[i]=0;
                head[i]=-1;
            }
            si=0;
            for(int i=1,u,v;i<N;i++){
                read(u);
                read(v);
                add_edge(u,v);
                add_edge(v,u);
            }
            scanf("%s%s",C,S);
            Pre[0]=0;
            Suf[0]=0;
            L=strlen(S);
            for(int i=0;i<L;i++)Pre[i+1]=Pre[i]*base+S[i];
            reverse(S,S+L);
            for(int i=0;i<L;i++)Suf[i+1]=Suf[i]*base+S[i];
          //  printf("target:%llu")
            int ans=solve(1);
            printf("Case #%d: ",cas);
          //  printf("___%d\n",ans);
            if(ans){
                puts("Find");
            }
            else puts("Impossible");
        }
        return 0;
    }