Submission #930638

#	Time	Username	Problem	Language	Result	Execution time	Memory
930638		LucaIlie	Fancy Fence (CEOI20_fancyfence)	C++17	30 / 100	61 ms	3792 KiB

fancyfence

#include <bits/stdc++.h>

using namespace std;

const int MAX_N = 1e5;
const int MOD = 1e9 + 7;
int h[MAX_N], w[MAX_N], p[MAX_N];
vector <int> s;

int lgPut( int x, int n ) {
    if ( n == 0 )
        return 1;

    int p = lgPut( x, n / 2 );
    p = (long long)p * p % MOD;
    if ( n % 2 == 1 )
        p = (long long)p * x % MOD;

    return p;
}

int main() {
    int n;

    cin >> n;
    for ( int i = 0; i < n; i++ )
        cin >> h[i];
    for ( int i = 0; i < n; i++ )
        cin >> w[i];


    int inv2 = lgPut( 2, MOD - 2 );
    int ans = 0, sum = 0;
    for ( int i = 0; i < n; i++ ) {
        while ( !s.empty() && h[i] <= h[s.back()] ) {
            sum = (sum - (long long)h[i] * (h[i] + 1) % MOD * inv2 % MOD * p[s.back()] % MOD + MOD) % MOD;
            p[i] = (p[i] + p[s.back()]) % MOD;
            s.pop_back();
        }
        s.push_back( i );

        sum = (sum + (long long)h[i] * (h[i] + 1) % MOD * inv2 % MOD * p[i] % MOD) % MOD;

        ans = (ans + ((long long)h[i] * (h[i] + 1) % MOD * inv2 % MOD) * ((long long)w[i] * (w[i] + 1) % MOD * inv2 % MOD)) % MOD;

        ans = (ans + (long long)sum * w[i] % MOD) % MOD;

        sum = (sum + (long long)h[i] * (h[i] + 1) % MOD * inv2 % MOD * w[i] % MOD) % MOD;

       // sum = (sum + w[i]) % MOD;
    }

    cout << ans << "\n";

    return 0;
}

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