Submission #745928

#	Time	Username	Problem	Language	Result	Execution time	Memory
745928		vjudge1	Fancy Fence (CEOI20_fancyfence)	C++17	0 / 100	2 ms	284 KiB

fancyfence

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


int main(){
    int n; cin >> n;
    int x, e, hossz;
    int ans = 0;
    vector<int> w, h;
    int wsum = 0;
    for(int i = 0; i < n; i++){
        cin >> x;
        h.push_back(x);
    }

    for(int i = 0; i < n; i++){
        cin >> x;
        wsum += x;
        w.push_back(x);
    }

    hossz = (h[0]-1)*w[0];
    for(int i = 1; i < n; i++){
        if(h[i] == 2){
            hossz += w[i];
        }
        else if(h[i-1] == 2){
     //       cout << hossz << " ";
            ans += hossz*(hossz+1);
            ans = ans % 1000000007;
            hossz = 0;
        }
    }
 //   cout << hossz << " ";
    ans += hossz*(hossz+1); 
    ans = ans % 1000000007;
    
    ans += wsum*(wsum+1)/2;
    ans = ans % 1000000007;

    cout << ans;
}

Compilation message (stderr)

fancyfence.cpp: In function 'int main()':
fancyfence.cpp:7:12: warning: unused variable 'e' [-Wunused-variable]
    7 |     int x, e, hossz;
      |            ^

• 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