#558675 (C++11) No.1239 Multiplication -2

提出ソース

結果

問題	No.1239 Multiplication -2
ユーザー	👑tozangezan
提出日時	2020-09-26 00:09:38
言語	C++11 (gcc 11.4.0)
結果	AC
実行時間	131 ms / 2,000 ms
コード長	3,341 bytes
コンパイル時間	743 ms
コンパイル使用メモリ	96,736 KB
実行使用メモリ	31,296 KB
最終ジャッジ日時	2023-09-10 17:17:00
合計ジャッジ時間	3,621 ms
ジャッジサーバーID （参考情報）	judge12 / judge14

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	19 ms 21,128 KB
testcase_01	AC	19 ms 21,140 KB
testcase_02	AC	19 ms 21,068 KB
testcase_03	AC	19 ms 21,140 KB
testcase_04	AC	19 ms 21,140 KB
testcase_05	AC	19 ms 21,140 KB
testcase_06	AC	19 ms 21,076 KB
testcase_07	AC	18 ms 21,132 KB
testcase_08	AC	19 ms 21,136 KB
testcase_09	AC	18 ms 21,132 KB
testcase_10	AC	19 ms 21,072 KB
testcase_11	AC	18 ms 21,132 KB
testcase_12	AC	18 ms 21,064 KB
testcase_13	AC	19 ms 21,064 KB
testcase_14	AC	19 ms 21,068 KB
testcase_15	AC	28 ms 21,388 KB
testcase_16	AC	34 ms 21,528 KB
testcase_17	AC	36 ms 21,464 KB
testcase_18	AC	37 ms 21,464 KB
testcase_19	AC	38 ms 21,468 KB
testcase_20	AC	39 ms 21,544 KB
testcase_21	AC	131 ms 31,296 KB
testcase_22	AC	43 ms 21,524 KB
testcase_23	AC	73 ms 27,936 KB
testcase_24	AC	39 ms 21,528 KB
testcase_25	AC	31 ms 21,528 KB
testcase_26	AC	94 ms 29,208 KB
testcase_27	AC	37 ms 23,632 KB
testcase_28	AC	81 ms 29,260 KB
testcase_29	AC	94 ms 29,960 KB
testcase_30	AC	33 ms 21,388 KB
testcase_31	AC	39 ms 21,532 KB
testcase_32	AC	53 ms 21,532 KB
testcase_33	AC	43 ms 21,600 KB
testcase_34	AC	41 ms 21,536 KB
testcase_35	AC	30 ms 21,380 KB
testcase_36	AC	29 ms 21,368 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

// #pragma GCC target("avx")  // CPU 処理並列化
// #pragma GCC optimize("O3")  // CPU 処理並列化
// #pragma GCC optimize("unroll-loops")  // 条件処理の呼び出しを減らす
// #define BEGIN_STACK_EXTEND(size) void * stack_extend_memory_ = malloc(size);void * stack_extend_origin_memory_;char * stack_extend_dummy_memory_ = (char*)alloca((1+(int)(((long long)stack_extend_memory_)&127))*16);*stack_extend_dummy_memory_ = 0;asm volatile("mov %%rsp, %%rbx\nmov %%rax, %%rsp":"=b"(stack_extend_origin_memory_):"a"((char*)stack_extend_memory_+(size)-1024));
// #define END_STACK_EXTEND asm volatile("mov %%rax, %%rsp"::"a"(stack_extend_origin_memory_));free(stack_extend_memory_);

#include<stdio.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<deque>
#include<stack>
#include<string>
#include<string.h>
#include<vector>
#include<set>
#include<map>
#include<bitset>
#include<stdlib.h>
#include<cassert>
#include<time.h>
#include<bitset>
#include<numeric>
#include<unordered_set>
#include<unordered_map>
#include<complex>
using namespace std;
const long long mod=998244353;
const long long inf=mod*mod;
const long long d2=(mod+1)/2;
const double EPS=1e-11;
const double INF=1e+10;
const double PI=acos(-1.0);
const int C_SIZE = 11100000;
const int UF_SIZE = 3100000;
namespace{
	long long fact[C_SIZE];
	long long finv[C_SIZE];
	long long inv[C_SIZE];
	inline long long Comb(int a,int b){
	 	if(a<b||b<0)return 0;
	 	return fact[a]*finv[b]%mod*finv[a-b]%mod;
	}
	void init_C(int n){
		fact[0]=finv[0]=inv[1]=1;
		for(int i=2;i<n;i++){
			inv[i]=(mod-(mod/i)*inv[mod%i]%mod)%mod;
		}
		for(int i=1;i<n;i++){
			fact[i]=fact[i-1]*i%mod;
			finv[i]=finv[i-1]*inv[i]%mod;
		}
	}
	long long pw(long long a,long long b){
		if(a<0LL)return 0;
		if(b<0LL)return 0;
		long long ret=1;
		while(b){
			if(b%2)ret=ret*a%mod;
			a=a*a%mod;
			b/=2;
		}
		return ret;
	}
	long long pw_mod(long long a,long long b,long long M){
		if(a<0LL)return 0;
		if(b<0LL)return 0;
		long long ret=1;
		while(b){
			if(b%2)ret=ret*a%M;
			a=a*a%M;
			b/=2;
		}
		return ret;
	}
	int pw_mod_int(int a,int b,int M){
		if(a<0)return 0;
		if(b<0)return 0;
		int ret=1;
		while(b){
			if(b%2)ret=(long long)ret*a%M;
			a=(long long)a*a%M;
			b/=2;
		}
		return ret;
	}
	int ABS(int a){return max(a,-a);}
	long long ABS(long long a){return max(a,-a);}
	double ABS(double a){return max(a,-a);}
	int sig(double r) { return (r < -EPS) ? -1 : (r > +EPS) ? +1 : 0; }
	int UF[UF_SIZE];
	void init_UF(int n){
		for(int i=0;i<n;i++)UF[i]=-1;
	}
	int FIND(int a){
		if(UF[a]<0)return a;
		return UF[a]=FIND(UF[a]);
	}
	void UNION(int a,int b){
		a=FIND(a);b=FIND(b);if(a==b)return;
		if(UF[a]>UF[b])swap(a,b);
		UF[a]+=UF[b];UF[b]=a;
	}
}
// ここから編集しろ
map<pair<int,int>,long long>M;
int p[210000];
long long p2[210000];
int main(){
	int a;scanf("%d",&a);
	init_C(410000);
	p2[0]=1;
	for(int i=1;i<210000;i++){
		p2[i]=p2[i-1]*2%mod;
	}
	long long ret=0;
	for(int i=0;i<a;i++){
		scanf("%d",p+i);
	}
	M[make_pair(0,0)]++;
	int ng=0;
	int tw=0;
	for(int i=0;i<a;i++){
		if(p[i]==0){
			M.clear();
			ng=0;
			tw=0;
		}else{
			if(p[i]<0)ng=!ng;
			if(ABS(p[i])==2)tw++;
			long long tmp=M[make_pair(!ng,tw-1)]%mod;
			ret=(ret+tmp*p2[max(0,a-2-i)])%mod;
		}
		M[make_pair(ng,tw)]+=p2[i];
	}
	ret=ret*pw(d2,a-1)%mod;
	printf("%lld\n",ret);
}

yukicoder

結果

テストケース

ソースコード