Submission #745987

#	Time	Username	Problem	Language	Result	Execution time	Memory
745987		vjudge1	Fancy Fence (CEOI20_fancyfence)	C++17	15 / 100	75 ms	2016 KiB

fancyfence

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

const int mod=1000000007;

int f(int wossz, int hossz){
	return (((((hossz*(hossz+1)/2)%mod) * wossz)%mod)*(wossz+1)/2)%mod;
}

int32_t main() {
	// all hi are equal --> téglalap
	int s;
	cin>>s;
	int h[s];
	int w[s];
	int wossz = 0;
	bool teglalap = true;
	for (int i=0; i<s; i++){
		cin >> h[i];
		if (h[i] != h[0]) teglalap = false;
	}
	int hossz = h[0];
	for(int i=0; i<s; i++){
		cin>>w[i];
		wossz = (w[i] + wossz) % mod;
	}
	
	if (teglalap)	cout << f(wossz, hossz);
	//cout << (((((hossz*(hossz+1)/2)%mod) * wossz)%mod)*(wossz+1)/2)%mod<<"\n";
	// --------------- lépcső -----------------
	/*int ans = f(wossz, h[0]);
	wossz -= w[0];
	for (int i=1; i<s; i++){
		ans += f(wossz, h[i]);
		ans -=  f(wossz, h[i-1]);
		wossz -= w[i]; 
	}
	cout << ans;*/

	return 0;
}

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