提出 #71366614

---

ソースコード 拡げる

#include<bits/stdc++.h>
#define cmin(a,b) a=std::min(a,b)
#define cmax(a,b) a=std::max(a,b)
#define inf
#define mod
#define maxn 1000006
#define int long long
int n,a[maxn];
char s[maxn],ch[maxn];
struct info{
	int x,c;
}ls[maxn];
int len[maxn],tot;
info operator +(info x,info y){
	if(x.x==y.x) return {x.x,x.c+y.c};
	return (x.x>y.x)?x:y; 
}
signed main(){
	scanf("%lld",&n);
	scanf("%s",s+1);
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
	} 
	char lst='?';info res={0,0}; 
	int mx=0,l=0,_c=0;
	for(int i=1;i<=n;i++){
		if(s[i]!=lst){
			if(lst!='?') ls[++tot]=res,ch[tot]=lst,len[tot]=l;
			lst=s[i],res={0,0},l=0;
		}
		res=res+(info){a[i],1},l++;
	}
	ls[++tot]=res,ch[tot]=lst,len[tot]=l;
	for(int i=1,l=1;i<=tot;l+=len[i],i++){
		if(ch[i]=='#'&&(i==1||i==tot)) continue;
		for(int j=l;j<=l+len[i]-1;j++){
			cmax(mx,a[j]);
		}
	}
	for(int i=2;i<=tot-1;i++){
		if(ch[i]=='#') _c++;
	}
	int ans=0;
	if(_c==1){
		for(int i=2;i<=tot-1;i++){
			if(ch[i]=='#') ans=ls[i-1].c*ls[i+1].c;
		}
	}else{
		for(int i=1;i<=tot;i++){
			if(ch[i]=='#') continue;
			if(i+2>tot) continue;
			if(ls[i+1].x==mx||ls[i].x==mx||ls[i+2].x==mx){
				ans+=ls[i].c*ls[i+2].c;
			}
		}
	}
	printf("%lld\n",ans);
	return 0;
}
/*
12
#...###...##
3 1 2 2 2 1 2 1 1 2 5 5
12
#...#.#...##
3 1 2 2 2 1 2 1 1 2 5 5
12
#...#..#...#
3 1 2 2 1 2 1 1 1 1 2 5 
*/

提出情報

ジャッジ結果