結果

問題 No.2129 Perfect Binary Tree...?
ユーザー kotatsugamekotatsugame
提出日時 2022-11-18 22:57:39
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 27 ms / 3,000 ms
コード長 1,255 bytes
コンパイル時間 913 ms
コンパイル使用メモリ 76,904 KB
実行使用メモリ 10,440 KB
最終ジャッジ日時 2023-10-20 07:56:11
合計ジャッジ時間 2,857 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
4,688 KB
testcase_01 AC 2 ms
4,688 KB
testcase_02 AC 3 ms
4,688 KB
testcase_03 AC 4 ms
4,688 KB
testcase_04 AC 6 ms
4,816 KB
testcase_05 AC 3 ms
4,688 KB
testcase_06 AC 10 ms
5,816 KB
testcase_07 AC 10 ms
5,032 KB
testcase_08 AC 15 ms
7,604 KB
testcase_09 AC 14 ms
6,908 KB
testcase_10 AC 2 ms
4,692 KB
testcase_11 AC 5 ms
5,144 KB
testcase_12 AC 13 ms
6,736 KB
testcase_13 AC 26 ms
10,440 KB
testcase_14 AC 19 ms
7,688 KB
testcase_15 AC 18 ms
7,032 KB
testcase_16 AC 22 ms
8,596 KB
testcase_17 AC 6 ms
4,820 KB
testcase_18 AC 4 ms
4,688 KB
testcase_19 AC 6 ms
4,828 KB
testcase_20 AC 10 ms
4,892 KB
testcase_21 AC 6 ms
4,848 KB
testcase_22 AC 7 ms
4,876 KB
testcase_23 AC 22 ms
9,164 KB
testcase_24 AC 22 ms
9,608 KB
testcase_25 AC 11 ms
6,180 KB
testcase_26 AC 4 ms
4,688 KB
testcase_27 AC 9 ms
5,924 KB
testcase_28 AC 16 ms
7,976 KB
testcase_29 AC 18 ms
8,480 KB
testcase_30 AC 27 ms
10,440 KB
testcase_31 AC 15 ms
6,908 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:21:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   21 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<atcoder/modint>
#include<cassert>
using namespace std;
using mint=atcoder::modint998244353;
int N;
string U,V;
mint ans;
mint p2[2<<17];
vector<mint>calc(string u,int idx)
{
	vector<mint>A;
	for(int i=u.size();i>idx;i--)
	{
		if(A.empty())A.push_back(p2[N-i+1]-1);
		else A.push_back(p2[N-i]);
	}
	return A;
}
main()
{
	cin>>N>>U>>V;
	p2[0]=1;
	for(int i=1;i<=N;i++)p2[i]=p2[i-1]+p2[i-1];
	for(int i=1;i<N;i++)ans+=p2[i]*(p2[N-i]-1)*(p2[N]-p2[N-i]);
	if(U==V)
	{
		cout<<ans.val()<<endl;
		return 0;
	}
	int idx=0;
	while(idx<U.size()&&idx<V.size()&&U[idx]==V[idx])idx++;
	vector<mint>A=calc(U,idx),B=calc(V,idx);
	if(A.empty()||B.empty())
	{
		A.push_back(p2[N]-1-(p2[N-idx+1]-1)+1);
		A.back()+=p2[N-idx]-1;
	}
	else
	{
		A.push_back(p2[N]-1-(p2[N-idx+1]-1)+1);
	}
	while(!B.empty())
	{
		A.push_back(B.back());
		B.pop_back();
	}
	mint s=0;
	for(mint a:A)s+=a;
	assert(s==p2[N]-1);
	vector<mint>SA1(A.size()),SA2(A.size());
	for(int j=A.size();j--;)
	{
		if(j+1<A.size())SA1[j]=SA1[j+1],SA2[j]=SA2[j+1];
		SA1[j]+=((int)A.size()-2*j)*A[j];
		SA2[j]+=A[j];
	}
	for(int i=0;i<A.size();i++)
	{
		int lj=i+A.size()/2+1;
		if(lj<A.size())
		{
			ans+=A[i]*(SA1[lj]+SA2[lj]*2*i);
		}
	}
	cout<<ans.val()<<endl;
}
0