其他分享
首页 > 其他分享> > P5016 龙虎斗

P5016 龙虎斗

作者:互联网

        思路:定义一个一维数组,存储每个兵营所含的人数,假设这个数组为a数组,100%的数据满足 n≤10^5;所以 a[100010];然后,再求出龙虎双方的势力:

1 for(int i=1;i<m;++i)
2 {
3     p+=(m-i)*a[i];
4 }
5 for(int i=m+1;i<=n;++i)
6 {
7     q+=(i-m)*a[i];
8 }

 

        p代表龙方的势力,q代表虎方的势力,稳妥一些,开long long比较好;

        然后再求出两方的势力差(要正数,用abs函数取绝对值就可以)然后就是从1到n循环一次,循环时,我分成了三种情况:

 1 for(int i=1;i<=n;++i)
 2     {
 3         if(i<m)
 4         {
 5             p+=s*(m-i);
 6             w=abs(p-q);
 7             if(w<t)
 8             {
 9                 t=w;
10                 ans=i;
11             }
12             p=p1;
13         }
14         if(i==m)
15         {
16             w=abs(p-q);
17             if(w<t)
18             {
19                 t=w;
20                 ans=i;
21             }
22         }
23         if(i>m)
24         {
25             q+=s*(i-m);
26             w=abs(p-q);
27             if(w<t)
28             {
29                 t=w;
30                 ans=i;
31             }
32             q=q1;
33         }
34     }

        t为两方的势力差+1,p1和q1分别为p和q的备份。当i<m时,就相当于增加龙方的势力,然后将它所增加的势力算出来,加到p上,再计算p与q的差,判断相比于之前是否差值变小,变小的话就将t更新为现在的差,并记录数组的位置。另外两种情况也类似。这样跑一遍,ans就是最终的答案。

        在此之前我还进行了一次特判,如果我手里的工兵为0的话,就直接输出1,因为没有工兵的话改变不了两方的差值,并且输出要求如果存在多个编号同时满足最优,取最小的编号。

完整代码:

 1 #include<iostream>
 2 #include<cmath>
 3 using namespace std;
 4 int long long a[100010];
 5 int main()
 6 {
 7     long long n,p=0,q=0,p1,q1,ans,w,t;
 8     cin>>n;
 9     for(int i=1;i<=n;++i)
10     {
11         cin>>a[i];
12     }
13     long long m,z,k,s;
14     cin>>m>>z>>k>>s;
15     if(s==0)
16     {
17         cout<<"1";
18         return 0;
19     }
20     a[z]+=k;
21     for(int i=1;i<m;++i)
22     {
23         p+=(m-i)*a[i];
24     }
25     for(int i=m+1;i<=n;++i)
26     {
27         q+=(i-m)*a[i];
28     }
29     p1=p,q1=q;
30     t=abs(p-q)+1;
31     for(int i=1;i<=n;++i)
32     {
33         if(i<m)
34         {
35             p+=s*(m-i);
36             w=abs(p-q);
37             if(w<t)
38             {
39                 t=w;
40                 ans=i;
41             }
42             p=p1;
43         }
44         if(i==m)
45         {
46             w=abs(p-q);
47             if(w<t)
48             {
49                 t=w;
50                 ans=i;
51             }
52         }
53         if(i>m)
54         {
55             q+=s*(i-m);
56             w=abs(p-q);
57             if(w<t)
58             {
59                 t=w;
60                 ans=i;
61             }
62             q=q1;
63         }
64     }
65     cout<<ans;
66     return 0;
67 }

 

标签:势力,int,两方,long,abs,数组,P5016,龙虎斗
来源: https://www.cnblogs.com/zkw666/p/12724946.html