4分蒟蒻求助大佬
查看原帖
4分蒟蒻求助大佬
742524
zhangchu楼主2023/7/19 23:38

本蒟蒻太菜,代码很乱望大佬体谅

#include<bits/stdc++.h>
using namespace std;
long long n,m,p1,p2,s1,s2,a1,a2,xxx,u,f,aa1;
long long s[100000];
int main(){
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>s[i];
	}
	cin>>m>>p1>>s1>>s2;
	s[p1]+=s1;
	for(int i=0;i<m;i++){
		a1+=(m-i)*s[i];
	}
	for(int i=m+1;i<n;i++){
		a2+=(i-m)*s[i];
	}
	if(a1<a2){
		xxx=a2-a1;
		u=xxx;
		aa1=a1;
		for(int i=0;i<m;i++){
		a1=aa1;
		a1+=s2*(m-i);
		if(u>xxx-a2+a1){
			u=xxx-a2+a1;
			f=i;
		}
	}
	if(u>xxx){
		cout<<m;
	}else{
		cout<<s[f];
	}
	}else if(a1>a2){
		xxx=a1-a2;
		u=xxx;
		aa1=a2;
		for(int i=m+1;i<n;i++){
			a2=aa1;
		a2+=s2*(m-i);
		if(u>xxx-a1+a2){
			u=xxx-a1+a2;
			f=i;
		}
	}
	if(u>xxx){
		cout<<m;
	}else{
		cout<<s[f];
	}
	}else{
		cout<<m;
	}
	return 0;
}
2023/7/19 23:38
加载中...