Submission #734285

#	Time	Username	Problem	Language	Result	Execution time	Memory
734285		josanneo22	Rainy Markets (CCO22_day1problem2)	C++17	0 / 25	275 ms	76664 KiB

Main

#include<bits/stdc++.h>
using namespace std;
#define int long long
void solve() { 
	int n; cin>>n;
	vector<int> b(n),p(n-1),u(n-1);
	for(auto&x:b) cin>>x;
	for(auto&x:p) cin>>x;
	for(auto&x:u) cin>>x;
	int ans[n][3]; memset(ans,0,sizeof(ans));
	int dp[n+1];memset(dp,0,sizeof(dp));
	int ok=1;
	for(int i=0;i<n-1;i++){
		int left=b[i]-dp[i];
		if(left>=p[i]){
			dp[i]+=p[i];
			p[i]=0;
			ans[i][0]=p[i];
		}
		else{
			dp[i]+=left;
			ans[i][0]=left;
			p[i]-=left;
			left=b[i+1];
			if(left>=p[i]){
				dp[i+1]+=p[i];
				ans[i][2]=p[i];
				p[i]=0;
			}
			else{
				ok=0; 
			}
		}
	}
	if(ok){
		cout<<"YES\n";
		cout<<0<<'\n';
		for(int i=0;i<n-1;i++){
			cout<<ans[i][0]<<' '<<ans[i][1]<<' '<<ans[i][2]<<'\n';
		}
	}
	else{
		cout<<"NO\n";
	}
}
signed main() {
	ios_base::sync_with_stdio(0); cin.tie(0);
	int tt = 1;
	while (tt--) solve();
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 5
• Subtask 3: 0 / 6
• Subtask 4: 0 / 9