제출 #745998

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
745998		vjudge1	Event Hopping (BOI22_events)	C++17	0 / 100	83 ms	1496 KiB

events

// Source: https://usaco.guide/general/io

#include <bits/stdc++.h>
using namespace std;

const long long MOD=1e9+7;

int main() {
	long long n;
	cin>>n;
	vector<long long> h(n);
	long long wossz=0, ans=0;
	for(long long i=0; i<n; i++){
		cin>>h[i];
	}
	vector<long long> ket;
	long long a;
	cin>>a;
	if(h[0]==2) ket.push_back(a);
	wossz+=a;
	for(long long i=1; i<n; i++){
		cin>>a;
		if(h[i]==h[i-1] && h[i]==2){
			ket.back()=(ket.back()+a)%MOD;
		}
		else if(h[i]==2){
			ket.push_back(a);
		}
		wossz=(wossz+a)%MOD;
	}
	for(long long x : ket){
		ans=(ans+(x*(x+1)%MOD))%MOD;
	}
	ans=(ans+(wossz*(wossz+1)/2)%MOD)%MOD;
	cout<<ans<<"\n";
}

• Subtask 1: 0 / 10
• Subtask 2: 0 / 10
• Subtask 3: 0 / 15
• Subtask 4: 0 / 15
• Subtask 5: 0 / 20
• Subtask 6: 0 / 30