結果

問題 No.1503 Bitwise And Convolution Twisted
ユーザー kotatsugamekotatsugame
提出日時 2021-05-07 22:24:21
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,040 ms / 2,000 ms
コード長 795 bytes
コンパイル時間 791 ms
コンパイル使用メモリ 67,936 KB
実行使用メモリ 19,968 KB
最終ジャッジ日時 2024-09-15 10:27:36
合計ジャッジ時間 7,061 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 13 ms
19,712 KB
testcase_01 AC 14 ms
19,840 KB
testcase_02 AC 14 ms
19,840 KB
testcase_03 AC 14 ms
19,712 KB
testcase_04 AC 15 ms
19,840 KB
testcase_05 AC 14 ms
19,712 KB
testcase_06 AC 30 ms
19,840 KB
testcase_07 AC 271 ms
19,840 KB
testcase_08 AC 30 ms
19,712 KB
testcase_09 AC 1,033 ms
19,968 KB
testcase_10 AC 1,040 ms
19,712 KB
testcase_11 AC 1,035 ms
19,840 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:7:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
    7 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<atcoder/modint>
using namespace std;
using mint=atcoder::modint998244353;
int N;
mint A[1<<20],B[1<<20],S[1<<20],T[1<<20];
main()
{
	cin>>N;
	for(int i=0;i<1<<N;i++)
	{
		int a;cin>>a;
		A[i]=a;
	}
	for(int i=0;i<1<<N;i++)
	{
		int b;cin>>b;
		B[i]=b;
	}
	for(int i=0;i<1<<N;i++)S[i]=A[i];
	for(int k=0;k<N;k++)for(int i=0;i<1<<N;i++)
	{
		if(i>>k&1)S[i^1<<k]+=S[i];
	}
	for(int i=0;i<1<<N;i++)
	{
		T[i]=B[i];
		if(__builtin_parity(i))T[i]=-T[i];
	}
	for(int k=0;k<N;k++)for(int i=0;i<1<<N;i++)
	{
		if(!(i>>k&1))T[i|1<<k]+=T[i];
	}
	for(int i=0;i<1<<N;i++)
	{
		if(__builtin_parity(i))T[i]=-T[i];
		S[i]*=T[i];
	}
	for(int k=0;k<N;k++)for(int i=0;i<1<<N;i++)
	{
		if(!(i>>k&1))S[i|1<<k]+=S[i];
	}
	for(int i=0;i<1<<N;i++)cout<<S[i].val()<<(i+1==(1<<N)?"\n":" ");
}
0