蒟蒻76分求助啊

下面是丑的不行的代码 #include <bits/stdc++.h> #define ll long long using namespace std; //qishi e[100001]; ll n,m,p1,p2,s1,s2,c = INT_MAX; ll a[100001],dr,ti; int main(){ cin >> n; for (int i = 1;i <= n;++i)cin >> a[i]; cin >> m >> p1 >> s1 >> s2; a[p1]+=s1; for (int i = 1;i < m;++i)dr += a[i](m- i); for (int i = m + 1;i <= n;++i)ti+= a[i](i - m); p2 = m; if(dr < ti) for (int i = 1;i <= m;++i){ if(abs((dr + s2 * (m - i)) - ti) < abs(dr-ti)) { if(c > abs((dr + s2 * (m - i)) - ti) ) { p2 = i; c = abs((dr + s2 * (m - i)) - ti); } } } else if (dr > ti){ for (int i = m ;i <= n;++i){ if(abs(ti + s2 * (i - m) - dr) < abs(ti-dr)){ if(c > abs((ti + s2 * (m - i)) - dr )){ p2 = i; c = abs((dr + s2 * (m - i)) - ti); } } } // cout << p2 << ' '; } if (c > abs(dr - ti)) cout << m <<endl; else cout << p2 ; // else cout }