[Noip2018普及组]龙虎斗

最新推荐文章于 2024-06-23 22:40:05 发布

原创最新推荐文章于 2024-06-23 22:40:05 发布 · 568 阅读

0 ·

CC 4.0 BY-SA版权

本文提供了一段使用C++编写的代码，该代码旨在通过调整一组数值来最小化两组数值之间的差距。通过输入一系列数值和特定参数，程序将计算并输出能够使两组数值总和差距最小化的最佳数值位置。

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
long long a[maxn];
int main(){
    long long n;
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    long long m,p1,s1,s2;
    cin>>m>>p1>>s1>>s2;
    a[p1]+=s1;
    long long sum1=0;
    long long sum2=0;
    for(int i=1;i<=m-1;i++)
        sum1+=a[i]*(m-i);
    for(int i=m+1;i<=n;i++)
        sum2+=a[i]*(i-m);
    long long minn=abs(sum1-sum2);
    long long x=m;//即便是小于也有可能派出去使得双方悬殊变大,刚开始派到中立派
    if(sum1<sum2){//增加sum1;  
        for(int i=1;i<=m-1;i++){
            long long temp=sum1+s2*abs(i-m);
            if(abs(temp-sum2)<minn){
                minn=abs(temp-sum2);
                x=i;
            }
        }
    }
    else if(sum1>sum2){//增加sum2
        for(int i=m+1;i<=n;i++){
            long long temp=sum2+s2*abs(i-m);
            if(abs(temp-sum1)<minn){
                minn=abs(temp-sum1);
                x=i;
            }
        }
    }
    else
        x=m;
    cout<<x<<endl;
    return 0;
}