KeDEMj - Online C++0x Compiler & Debugging Tool

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
const int MOD = 158400;
const int MOD2 = 998244353;
const int INF = LLONG_MAX/2;

int primes[1000000];

void seive(){
    fill(primes, primes + 1000000, 1);
    primes[0] = primes[1] = 0;
    for(int i = 2 ; i*i < 1000000 ; i++){
        if(primes[i]){
            for(int j = i*i ; j < 1000000 ; j += i){
                primes[j] = 0;
            }
        }
    }
    for(int i = 1 ; i < 1000000 ; i++){
        primes[i] += primes[i-1];
    }
}
int factorial(int n){
    if(n==0){
        return 1;
    }
    return (n*(factorial(n-1)))%MOD;
}
bool isPrime(int n){
    if(n <= 1) return false;
    for(int i = 2 ; i*i <= n ; i++){
        if(n % i == 0) return false;
    }
    return true;
}

int power(int a, int b){
    if(b == 0) return 1;
    a %= MOD;
    int value = power(a, b / 2);
    if(b % 2 == 0){
        return (value * value) % MOD;
    } else {
        return ((value * value) % MOD * (a % MOD)) % MOD;
    }
}

int gcd(int a, int b){
    if(a == 0) return b;
    return gcd(b % a, a);
}
void solve() {
    int n,power;
    cin>>n>>power;
    int damage[n],health[n];
    for(int i = 0 ; i<n ; i++){
        cin>>damage[i];
    }
    for(int i = 0 ; i<n ; i++){
        cin>>health[i];
    }
    for(int i = 0 ; i<n ; i++){
        health[i] = (health[i]+power-1)/power;
    }
    vector<pair<double,int>>v;
    int sum = 0;
    for(int i = 0 ; i<n ; i++){
        double f = double(damage[i])/double(health[i]);
        v.push_back({f, i});
        sum += damage[i];
    }
    sort(v.begin(),v.end());
    int sum2 = 0;
    for(int i = n-1 ; i>=0 ; i--){
        int index = v[i].second;
        sum2 += (sum)*health[index];
        sum -= damage[index];
    }
    cout<<sum2<<endl;
}

signed main(){
    ios::sync_with_stdio(false); cin.tie(NULL);
    //int t;
    //cin >> t;
    //while(t--){
        solve();
    //}
    return 0;
}