結果

問題 No.263 Common Palindromes Extra
ユーザー msm1993msm1993
提出日時 2020-11-12 09:17:00
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 239 ms / 2,000 ms
コード長 2,479 bytes
コンパイル時間 2,150 ms
コンパイル使用メモリ 182,792 KB
実行使用メモリ 161,404 KB
最終ジャッジ日時 2024-07-22 19:03:09
合計ジャッジ時間 3,641 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 8 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 13 ms
6,272 KB
testcase_04 AC 55 ms
14,728 KB
testcase_05 AC 80 ms
14,452 KB
testcase_06 AC 7 ms
5,376 KB
testcase_07 AC 102 ms
50,624 KB
testcase_08 AC 130 ms
50,284 KB
testcase_09 AC 144 ms
91,188 KB
testcase_10 AC 239 ms
161,404 KB
testcase_11 AC 52 ms
14,468 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;

#define int long long

#define rep(i,n) for(int i=0;i<(n);i++)
#define pb push_back
#define eb emplace_back
#define all(v) (v).begin(),(v).end()
#define fi first
#define se second
typedef vector<int>vint;
typedef pair<int,int>pint;
typedef vector<pint>vpint;

template<typename A,typename B>inline void chmin(A &a,B b){if(a>b)a=b;}
template<typename A,typename B>inline void chmax(A &a,B b){if(a<b)a=b;}

template<class A,class B>
ostream& operator<<(ostream& ost,const pair<A,B>&p){
	ost<<"{"<<p.first<<","<<p.second<<"}";
	return ost;
}

template<class T>
ostream& operator<<(ostream& ost,const vector<T>&v){
	ost<<"{";
	for(int i=0;i<v.size();i++){
		if(i)ost<<",";
		ost<<v[i];
	}
	ost<<"}";
	return ost;
}

/*
p->xpx  :  node(p).nex[x]=node_id(xpx)        (enlarge link)
aabaa->aa :  node(aabaa).suflink=node_id(aa)    (suffix link)
sufs[i] : node_id(longest suffix-palindrome of s[0,i])
*/
struct PalindromicTree{
    struct Node{
    	map<char,int>nex;
        int len,suflink,cnt;
		Node(int len,int suflink,int cnt)
			:len(len),suflink(suflink),cnt(cnt){}
    };

    string s;
    vector<Node>v;
	vector<int>sufs;
	int pos,suf;

	PalindromicTree():pos(0),suf(1){
		v.eb(-1,0,0);
		v.eb(0,0,0);
    }

	void add(const string &t){
		s+=t;
	}

    void uku(){
		assert(pos<s.size());
        char ch=s[pos];
        int cur=suf;
        while(true){
            if(pos-1-v[cur].len>=0&&s[pos-1-v[cur].len]==ch)break;
            cur=v[cur].suflink;
        }

        if(v[cur].nex.count(ch)){
            suf=v[cur].nex[ch];
            v[suf].cnt++;
		    return;
        }

        suf=v.size();
		v.eb(v[cur].len+2,-1,1);
        v[cur].nex[ch]=suf;

        if(v[suf].len==1){
            v[suf].suflink=1;
            return;
        }

        while(true){
            cur=v[cur].suflink;
            if(pos-1-v[cur].len>=0&&s[pos-1-v[cur].len]==ch){
                v[suf].suflink=v[cur].nex[ch];
				return;
            }
        }
    }

	void process(){
		uku();
		pos++;
		sufs.pb(suf);
	}
};

int dps[1111111],dpt[1111111];

signed main(){
	string S,T;
	cin>>S>>T;

	PalindromicTree pt;
	pt.add(S);
	pt.add("@*");
	pt.add(T);
	rep(i,pt.s.size())pt.process();
	rep(i,S.size())dps[pt.sufs[i]]++;
	rep(i,T.size())dpt[pt.sufs[S.size()+2+i]]++;
	int ans=0;
	for(int i=(int)pt.v.size()-1;i>=2;i--){
		ans+=dps[i]*dpt[i];
		dps[pt.v[i].suflink]+=dps[i];
		dpt[pt.v[i].suflink]+=dpt[i];
	}
	cout<<ans<<endl;
	return 0;
}
0