しろは的军训列队

代码

#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long LL;

const int N = 1e6;
int n , a[N + 5] , b[N + 5];
struct node{
    int a , b;
}c[N + 5];
LL val , ans , k1 , k2;

inline bool cmp(node x , node y) { return x.a < y.a; }

int main()
{
    scanf("%d" , &n);
    for(register int i = 1; i <= n; i++) scanf("%d" , &c[i].a);
    for(register int i = 1; i <= n; i++) scanf("%d" , &c[i].b);
    sort(c + 1 , c + n + 1 , cmp);
    for(register int i = 2; i <= n; i++) val += 1LL * (c[i].a - c[1].a) * c[i].b , k1 += 1LL * c[i].b;
    k2 = c[1].b , ans = val;
    for(register int i = 2; i <= n; i++)
    {
        val -= 1LL * k1 * (c[i].a - c[i - 1].a);
        val += 1LL * k2 * (c[i].a - c[i - 1].a);
        k1 -= c[i].b , k2 += c[i].b;
        ans = min(ans , val); 
    }
    printf("%lld" , ans);
}