Submission #944833

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

fancyfence

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

int n;
int w[100005];
int h[100005];
int ans = 0;
int width = 0;
int mod = 1000000000 + 7;

signed main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    
    cin >> n;
    for(int i = 0; i < n; i++)cin >> h[i];
        for(int i = 0; i < n; i++){
        cin >> w[i];
        width += w[i];
        // ans += (h[i]-1)*w[i]; // vertical rectangle for h=2
    }
    // cout << width << "\n";
    // cout << ans << "\n";
    if(width % 2 == 0){ // all rectangle for h=1
        ans += (width+1)*(width/2);
        ans %= mod;
    } else {
        ans += (width*(width-1)/2)+width;
        ans %= mod;
    }
    // cout << ans << "\n";
    width = 0;
    for(int i = 0; i < n; i++){
        if(h[i] > 1){
            width += w[i];
        } else {
            if(width != 0){
                if(width % 2 == 0){ // all rectangle for h=2
                    ans += (width+1)*(width/2)*2;
                    ans %= mod;
                } else {
                    ans += ((width*(width-1)/2)+width)*2;
                    ans %= mod;
                }
            }
            width = 0;
        }
        if(i == (n-1)){
            if(width != 0){
                if(width % 2 == 0){ // all rectangle for h=2
                    ans += (width+1)*(width/2)*2;
                    ans %= mod;
                } else {
                    ans += ((width*(width-1)/2)+width)*2;
                    ans %= mod;
                }
            }
            width = 0;
        }
    }

    cout << ans;

    return 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