[HAOI2016]找相同字符

题目大意：
　　给你两个字符串a和b，要求从a和b中各取出一个相等的子串，问不同的取法有多少种。

思路：
　　对于a串建立SAM，然后DP求出每个状态Right集合的大小。
　　然后把b串放进去匹配，对于每一个匹配到的结点p，它的每一个Right状态都可以匹配一个长度为tmp-s[s[p].link].len的串，那么将s[p].right*(tmp-s[s[p].link].len)计入答案。
　　统计每个状态被匹配的次数，并累加进它Parent中。
　　将s[top[i]].right*f[top[i]]*(s[top[i]].len-s[s[top[i]].link].len)统计进答案。
　　然后发现样例都过不了，不管什么数据都输出0。发现是匹配的时候的w直接用ch复制，而没有用idx（这样就导致b串一点都匹配不了）。
　　再交上去发现还是WA了，考虑两个字符串完全相等的情况，答案可以达到$displaystyle{sum_{i=1}^{n}}n^2$，显然要开long long，然后就过了。

#include<cstdio>
#include<cstring>
const int LEN=200001;
char s[LEN];
class SuffixAutomaton {
    private:
        static const int SIGMA_SIZE=26;
        struct State {
            int link,trans[SIGMA_SIZE],len,right,app;
        };
        State s[LEN<<1];
        int root,last;
        int len;
        int sz,newState(const int l) {
            sz++;
            s[sz].len=l;
            return sz;
        }
        int idx(const char ch) {
            return ch-'a';
        }
        void extend(const char ch) {
            const int w=idx(ch);
            int p=last,new_p=newState(s[last].len+1);
            s[new_p].right=1;
            while(p&&!s[p].trans[w]) {
                s[p].trans[w]=new_p;
                p=s[p].link;
            }
            if(!p) {
                s[new_p].link=root;
            } else {
                int q=s[p].trans[w];
                if(s[q].len==s[p].len+1) {
                    s[new_p].link=q;
                } else {
                    int new_q=newState(s[p].len+1);
                    memcpy(s[new_q].trans,s[q].trans,sizeof s[q].trans);
                    s[new_q].link=s[q].link;
                    s[q].link=s[new_p].link=new_q;
                    while(p&&s[p].trans[w]==q) {
                        s[p].trans[w]=new_q;
                        p=s[p].link;
                    }
                }
            }
            last=new_p;
        }
        int cnt[LEN],top[LEN<<1],f[LEN<<1];
        void tsort() {
            for(int i=1;i<=sz;i++) {
                cnt[s[i].len]++;
            }
            for(int i=len;i;i--) {
                cnt[i-1]+=cnt[i];
            }
            for(int i=1;i<=sz;i++) {
                top[cnt[s[i].len]--]=i;
            }
        }
    public:
        void build(char str[]) {
            len=strlen(str);
            root=last=newState(0);
            for(int i=0;str[i];i++) {
                extend(str[i]);
            }
        }
        long long query(char str[]) {
            tsort();
            for(int i=1;i<=sz;i++) {
                if(s[top[i]].link) {
                    s[s[top[i]].link].right+=s[top[i]].right;
                }
            }
            long long ans=0,n=strlen(str),tmp=0;
            int p=root;
            for(int i=0;i<n;i++) {
                const int w=idx(str[i]);
                if(s[p].trans[w]) {
                    p=s[p].trans[w];
                    tmp++;
                } else {
                    while(p&&!s[p].trans[w]) {
                        p=s[p].link;
                    }
                    if(!p) {
                        p=root;
                        tmp=0;
                    } else {
                        tmp=s[p].len+1;
                        p=s[p].trans[w];
                    }
                }
                s[p].app++,ans+=(long long)s[p].right*(tmp-s[s[p].link].len);
            }
            for(int i=1;i<=sz;i++) {
                f[s[top[i]].link]+=f[top[i]]+s[top[i]].app;
            }
            for(int i=1;i<sz;i++) {
                ans+=(long long)s[top[i]].right*f[top[i]]*(s[top[i]].len-s[s[top[i]].link].len);
            }
            return ans;
        }
};
SuffixAutomaton sam;
int main() {
    scanf("%s",s);
    sam.build(s);
    scanf("%s",s);
    printf("%lld
",sam.query(s));
    return 0;
}