제출 #1188322

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1188322		NoMercy	Fancy Fence (CEOI20_fancyfence)	C++20	0 / 100	0 ms	328 KiB

fancyfence

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

typedef long long ll;

const int mod = 1e9 + 7;

int32_t main () {

    ios_base::sync_with_stdio(0); 
    cin.tie(0); 
    cout.tie(0); 

    int N;
    cin >> N;
    vector<ll> h(N), w(N);
    for (int i = 0;i < N;i ++) cin >> h[i];
    for (int i = 0;i < N;i ++) cin >> w[i];

    ll sum = 0, res = 0;
    stack<array<ll, 2>> st;
    for (int i = 0;i < N;i ++) {
        ll cnt = w[i];
        while (st.size() > 0 && st.top()[0] >= h[i]) {
            cnt += st.top()[1];
            sum = (sum - (((st.top()[0] * (st.top()[0] + 1) / 2) * st.top()[1]) % mod) + mod) % mod;
            st.pop();
        }
        st.push({h[i], cnt});
        res = (res + (sum * cnt % mod)) % mod;
        res = (res + ((h[i] * (h[i] + 1) / 2) * (cnt * (cnt + 1) / 2) % mod)) % mod;
        sum = (sum + ((h[i] * (h[i] + 1) / 2) * cnt % mod)) % mod;
        // cout << res << " " << sum << " " << cnt << "\n";
    }
    cout << res << "\n";
}

• 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