[NOIP2018 普及组] 龙虎斗

KO王者bob

已于 2024-07-05 13:53:20 修改

阅读量424

点赞数 17

文章标签： c++ 算法

于 2024-07-05 13:53:12 首次发布

本文链接：https://blog.csdn.net/BobBoh/article/details/140182056

版权

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

int n,m,p1;
long long c[110000],s1,s2;

int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%lld",&c[i]);
	}
	scanf("%d%d%lld%lld",&m,&p1,&s1,&s2);
	c[p1]+=s1;
	long long sum1=0,sum2=0;
	for(int i=1;i<=n;i++){
		if(i<m){
			sum1+=c[i]*(m-i);
		}
		else{
			sum2+=c[i]*(i-m);
		}
	}
	long long ans=-1;
	int ansp=-1;
	for(int p2=1;p2<=n;p2++){
		long long t1=sum1,t2=sum2;
		if(p2<m){
			t1+=s2*(m-p2);
			}
			else{
				if(p2>m){
					t2+=s2*(p2-m);
				}
			}
			long long tmp=abs(t1-t2);
			if(ansp==-1||tmp<ans){
				ans=tmp;
				ansp=p2;
			}
	}
	printf("%d",ansp);
}

免费的

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

int n,m,p1;
long long c[110000],s1,s2;

int main(){
   scanf("%d",&n);
   for(int i=1;i<=n;i++){
       scanf("%lld",&c[i]);
   }
   scanf("%d%d%lld%lld",&m,&p1,&s1,&s2);
   c[p1]+=s1;
   long long sum1=0,sum2=0;
   for(int i=1;i<=n;i++){
       if(i<m){
           sum1+=c[i]*(m-i);
       }
       else{
           sum2+=c[i]*(i-m);
       }
   }
   long long ans=-1;
   int ansp=-1;
   for(int p2=1;p2<=n;p2++){
       long long t1=sum1,t2=sum2;
       if(p2<m){
           t1+=s2*(m-p2);
           }
           else{
               if(p2>m){
                   t2+=s2*(p2-m);
               }
           }
           long long tmp=abs(t1-t2);
           if(ansp==-1||tmp<ans){
               ans=tmp;
               ansp=p2;
           }
   }
   printf("%d",ansp);
}