oj.uz

답안 #448677

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
448677 2021-07-31T15:39:56 Z Khizri Fancy Fence (CEOI20_fancyfence) C++17
15 / 100
 42 ms 2148 KB 
#include <bits/stdc++.h>
using namespace std;
//------------------------------DEFINE------------------------------
//******************************************************************
#define IOS ios_base::sync_with_stdio(false); cin.tie(0),cout.tie(0)
#define ll long long
#define pb push_back		 
#define F first																 
#define S second 															 
#define INF 1e18
#define all(v) (v).begin(),(v).end()
#define rall(v) (v).rbegin(),(v).rend()
#define pii pair<int,int>
#define pll pair<ll,ll>
#define OK cout<<"Ok"<<endl;
#define MOD (ll)(1e9+7)
#define endl "\n"
//******************************************************************
//----------------------------FUNCTION------------------------------
//******************************************************************
ll gcd(ll a,ll b){
	if(a>b) swap(a,b);
	if(a==0) return a+b;
	return gcd(b%a,a);
}
ll lcm(ll a,ll b){
	return a/gcd(a,b)*b;
}
bool is_prime(ll n){
	ll k=sqrt(n);
	if(n==2) return true;
	if(n<2||n%2==0||k*k==n) return false;
	for(int i=3;i<=k;i+=2){
		if(n%i==0){
			return false;
		}
	}
	return true;
}
//*****************************************************************
//--------------------------MAIN-CODE------------------------------
const int mxn=2e5+5;
ll t=1,n,arr[mxn],h[mxn];
ll cal(ll k,ll sum){
	sum=sum%MOD;
	ll a=k;
	ll b=k+1;
	if(a%2==0){
		a/=2;
	}
	else{
		b/=2;
	}
	ll x=(a*b)%MOD;
	a=sum;
	b=sum+1;
	if(a%2==0){
		a/=2;
	}
	else{
		b/=2;
	}
	ll y=(a*b)%MOD;
	ll ans=(x*y)%MOD;
	return ans;
}
void solve(){
	ll ans=0;
	ll sum=0,k=0;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>h[i];
	}
	for(int i=1;i<=n;i++){
		cin>>arr[i];
		sum+=arr[i];
	}
	stack<pll>st;
	st.push({h[1],arr[1]});
	for(int i=2;i<=n;i++){
		ll x=0;
		while(st.size()&&h[i]<=st.top().F){
			pll xx=st.top();
			st.pop();
			ll y=h[i];
			if(st.size()){
				y=st.top().F;
			}
			st.push(xx);
			ll say=cal(st.top().F,st.top().S+x)-cal(y,st.top().S+x);
			if(say<0){
				say+=MOD;
			}
			ans=(ans+say)%MOD;
			x+=st.top().S;
			st.pop();
		}
		st.push({h[i],x+arr[i]});
	}
	vector<pll>vt;
	while(st.size()){
		vt.pb(st.top());
		st.pop();
	}
	vt.pb({0,0});
	reverse(all(vt));
	for(int i=1;i<vt.size();i++){
		ll say=cal(vt[i].F,sum)-cal(vt[i-1].F,sum);
		if(say<0){
			say+=MOD;
		}
		ans=(ans+say)%MOD;
		sum-=vt[i].S;
	}
	cout<<ans<<endl;
}
int main(){
	IOS;
	//cin>>t;
	while(t--){
		solve();
	}
	return 0;
}

Compilation message

fancyfence.cpp: In function 'void solve()':
fancyfence.cpp:107:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  107 |  for(int i=1;i<vt.size();i++){
      |              ~^~~~~~~~~~
fancyfence.cpp:69:11: warning: unused variable 'k' [-Wunused-variable]
   69 |  ll sum=0,k=0;
      |           ^

Subtask #1
0.0 / 0.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 332 KB Output is correct
2 Incorrect 1 ms 332 KB Output isn't correct

Subtask #2
0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 1 ms 204 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 204 KB Output is correct
2 Incorrect 1 ms 332 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #4
15.0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 316 KB Output is correct
2 Correct 4 ms 452 KB Output is correct
3 Correct 15 ms 1100 KB Output is correct
4 Correct 32 ms 1852 KB Output is correct
5 Correct 34 ms 1952 KB Output is correct
6 Correct 1 ms 204 KB Output is correct

Subtask #5
0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 204 KB Output is correct
2 Correct 1 ms 332 KB Output is correct
3 Correct 5 ms 460 KB Output is correct
4 Correct 15 ms 1100 KB Output is correct
5 Correct 31 ms 1860 KB Output is correct
6 Correct 42 ms 1924 KB Output is correct
7 Correct 1 ms 332 KB Output is correct
8 Correct 4 ms 992 KB Output is correct
9 Incorrect 21 ms 2148 KB Output isn't correct
10 Halted 0 ms 0 KB -

Subtask #6
0 / 18.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 324 KB Output is correct
2 Incorrect 1 ms 332 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #7
0 / 27.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 332 KB Output is correct
2 Incorrect 1 ms 332 KB Output isn't correct