結果

問題 No.2219 Re:010
コンテスト
ユーザー vjudge1
提出日時 2025-01-22 15:45:08
言語 C++17
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++17 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 14 ms / 2,000 ms
コード長 1,137 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,484 ms
コンパイル使用メモリ 210,288 KB
実行使用メモリ 11,904 KB
最終ジャッジ日時 2026-06-21 09:01:49
合計ジャッジ時間 2,543 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
#define int long long
using namespace std;
char s[200010];
int n,mod,ans,tmp,z[200010],two[200010],zz[200010],w[200010],ww[200010];
signed main(){
	mod=998244353*1ll;
	scanf("%s",s+1);
	n=strlen(s+1);
    for(int i=1;i<=n;i++){
    	if(s[i]=='?'){
    		w[i]=w[i-1]+1ll;
    		z[i]=z[i-1];
		}
		else if(s[i]=='0'){
			w[i]=w[i-1];
			z[i]=z[i-1]+1ll;
		}
		else{
			w[i]=w[i-1];
			z[i]=z[i-1];
		}
	}
	for(int i=n;i>=1;i--){
		if(s[i]=='?'){
    		ww[i]=ww[i+1]+1ll;
    		zz[i]=zz[i+1];
		}
		else if(s[i]=='0'){
			ww[i]=ww[i+1];
			zz[i]=zz[i+1]+1ll;
		}
		else{
			ww[i]=ww[i+1];
			zz[i]=zz[i+1];
		}
	}
	two[0]=1ll;
	for(int i=1;i<=n;i++) two[i]=(two[i-1]*2ll)%mod;
	for(int i=1;i<=n;i++){
		if(s[i]!='0'){
			tmp=two[w[i-1]+ww[i+1]]*((z[i-1]*zz[i+1])%mod)%mod;
			if(w[i-1]>0) tmp=(tmp+two[w[i-1]+ww[i+1]-1]*(w[i-1]*zz[i+1]%mod)%mod)%mod;																														
			if(ww[i+1]>0) tmp=(tmp+two[w[i-1]+ww[i+1]-1]*(ww[i+1]*z[i-1]%mod)%mod)%mod;
			if(w[i-1]>0&&ww[i+1]>0) tmp=(tmp+two[w[i-1]+ww[i+1]-2]*(w[i-1]*ww[i+1]%mod)%mod)%mod;
			ans=(ans+tmp)%mod;
		}
	}
	printf("%lld",ans%mod);
	return 0;
}
0