Submission #1008244

#	Time	Username	Problem	Language	Result	Execution time	Memory
1008244		kaysan	Fancy Fence (CEOI20_fancyfence)	C++17	0 / 100	1 ms	348 KiB

fancyfence

#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define pp pop_back
#define fi first
#define se second
#define lb lower_bound
#define ub upper_bound
#define pll pair<ll,ll> 
#define pii pair<int,int> 
#define vl vector<ll> 
#define nikah ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
const ll maxn = 1e5+7, modn1 = 1e9+7, modn2 = 998244353;
using namespace std;

// kalo kamu ngga emas nanti nadya merasa bersalah, ayo san demi nadya!!!
// Kalo emang soalnya susah, pindah soal dulu 
// Jangan sampe diem, pikirin sesuatu, pasti bisa!!!
// Cobain semua approach, jangan ngestuck di satu pemikiran
// Jangan overthinking, jangan panik, tenangin pikiranmu


ll t,n;
ll h[maxn], w[maxn];
vector<ll>v,b;

ll f (ll x) {
	return ((x * (x+1)) / 2) % modn1;
}

void solve () {
	v.pb(0); b.pb(0);
	cin>>n;
	for (ll i=1; i<=n; i++) {
		cin>>h[i];
	}
	ll sum = 0;
	for (ll i=1; i<=n; i++) {
		cin>>w[i];
		if (i>1) { 
			if (h[i] != h[i-1]) {
				v.pb(h[i-1]);
				b.pb(w[i-1]);
			} else {
				w[i] += w[i-1];
			}
		}
	}
	v.pb(h[n]); b.pb(w[n]);
	ll ans = 0, cur = 1, cnt = 0;
	bool inc = 0, dec = 0;
	for (ll i=1; i<v.size(); i++) {
		if (i == v.size() - 1) {
			if (dec) {
				ll cnt2 = 0;
				for (ll j=i; j>=cur; j--) {
					cnt2 += b[j];
					if (j == i) ans = (ans + ((f(b[j]) * f(v[j])) % modn1)) % modn1;
					else {
						ans = (ans + ((f(cnt2) * f(v[j])) % modn1)) % modn1;
						ans = (ans - ((f(cnt2 - b[j]) * f(v[j])) % modn1)) % modn1;
					}
				}
				dec = 0;
			} else {
				cnt += b[i];
				if (inc == 0) ans = (ans + ((f(b[i]) * f(v[i])) % modn1)) % modn1;
				else {
					ans = (ans + ((f(cnt) * f(v[i])) % modn1)) % modn1;
					ans = (ans - ((f(cnt - b[i]) * f(v[i])) % modn1)) % modn1;
				}
			}
		} else if (v[i] > v[i+1]) {
			if (dec) {
				ll cnt2 = 0;
				for (ll j=i; j>=cur; j--) {
					cnt2 += b[j];
					if (j == i) ans = (ans + ((f(b[j]) * f(v[j])) % modn1)) % modn1;
					else {
						ans = (ans + ((f(cnt) * f(v[j])) % modn1)) % modn1;
						ans = (ans - ((f(cnt - b[j]) * f(v[j])) % modn1)) % modn1;
					}
				}
				dec = 0;
			} else {
				cnt += b[i];
				if (inc == 0) ans = (ans + ((f(b[i]) * f(v[i])) % modn1)) % modn1;
				else {
					ans = (ans + ((f(cnt) * f(v[i])) % modn1)) % modn1;
					ans = (ans - ((f(cnt - b[i]) * f(v[i])) % modn1)) % modn1;
				}
			}
			inc = 1;
		} else {
			if (!dec) {
				cur = i;
				inc = 0;
				cnt = 0;
			}
			dec = 1;
		}
	//	cout<<i<<" : "<<ans<<endl;
	}
	while (ans < 0) ans += modn1;
	cout<<ans<<endl;
}           


int main () {
  nikah
  t = 1;
//  cin>>t;
  while (t--) {
    solve();
  //  cout<<endl;
  }
}

Compilation message (stderr)

fancyfence.cpp: In function 'void solve()':
fancyfence.cpp:52:16: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   52 |  for (ll i=1; i<v.size(); i++) {
      |               ~^~~~~~~~~
fancyfence.cpp:53:9: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   53 |   if (i == v.size() - 1) {
      |       ~~^~~~~~~~~~~~~~~
fancyfence.cpp:37:5: warning: unused variable 'sum' [-Wunused-variable]
   37 |  ll sum = 0;
      |     ^~~

• Subtask 1: 0 / 0
• Subtask 2: 0 / 12
• Subtask 3: 0 / 13
• Subtask 4: 0 / 15
• Subtask 5: 0 / 15
• Subtask 6: 0 / 18
• Subtask 7: 0 / 27