Particle accelerator

input/code.cpp: In function 'int main()':
input/code.cpp:39:26: warning: comparison of integer expressions of different signedness: 'long long unsigned int' and 'int' [-Wsign-compare]
   39 |             if (smallest == -1 || smallest > array[j].first) {
      |                 ~~~~~~~~~^~~~~

#include <bits/stdc++.h>

static unsigned long long calc(const long a, const long b) {
    if (a < b) {
        return static_cast<unsigned long long>(a) * (b - a);
    }
    return static_cast<unsigned long long>(b) * (a - b);
}

int main() {
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(nullptr);
    
    long long n;
    std::cin >> n;

    std::vector<long> mass(n);
    for (int i = 0; i < n; i++) {
        std::cin >> mass[i];
    }

    std::vector<std::pair<unsigned long long, std::pair<int, int>>> array(n - 1 );
    for (int i = 0; i < n - 1; i++) {
        array[i].second.first = mass[i];
        array[i].second.second = mass[i + 1];
        array[i].first = calc(mass[i],mass[i + 1]);
    }

    unsigned long long total = 0;
    for (int i = 0; i < n - 1; i++) {
        unsigned long long smallest = -1;
        int index = -1;
        int prev_ind = -1;
        int tmp_var = -1;
        for (int j = 0; j < n - 1; j++) {
            if (array[j].first == 0) {
                continue;
            }
            if (smallest == -1 || smallest > array[j].first) {
                smallest = array[j].first;
                prev_ind = tmp_var;
                index = j;
            }
            tmp_var = j;
        }
        total += smallest;
        if (prev_ind != -1) {
            array[prev_ind].second.second = array[index].second.first + array[index].second.second;
            array[prev_ind].first = calc(array[prev_ind].second.first, array[prev_ind].second.second);
        }
        array[index].first = 0;
        for (int j = index + 1; j < n-1; j++) {
            if (array[j].first != 0) {
                array[j].second.first = array[index].second.first + array[index].second.second;
                array[j].first = calc(array[j].second.first, array[j].second.second);
                break;
            }
        }
    }

    std::cout << total << std::endl;
    return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: WRONG ANSWER

Test 10

Verdict: WRONG ANSWER

Test 11

Verdict: WRONG ANSWER

Test 12

Verdict: WRONG ANSWER

Test 13

Verdict: ACCEPTED

Test 14

Verdict: WRONG ANSWER

Test 15

Verdict: WRONG ANSWER

Test 16

Verdict: ACCEPTED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: WRONG ANSWER

Test 19

Verdict: ACCEPTED

Test 20

Verdict: ACCEPTED

Test 21

Verdict: ACCEPTED

Test 22

Verdict: WRONG ANSWER

Test 23

Verdict: WRONG ANSWER

Test 24

Verdict: WRONG ANSWER

Test 25

Verdict: WRONG ANSWER

Test 26

Verdict: WRONG ANSWER

Test 27

Verdict: WRONG ANSWER

Test 28

Verdict: WRONG ANSWER

Test 29

Verdict: WRONG ANSWER

Test 30

Verdict: WRONG ANSWER

Test 31

Verdict: WRONG ANSWER

Test 32

Verdict: WRONG ANSWER

Test 33

Verdict: WRONG ANSWER

Test 34

Verdict: WRONG ANSWER

Test 35

Verdict: WRONG ANSWER

Test 36

Verdict: WRONG ANSWER

Test 37

Verdict: WRONG ANSWER

Test 38

Verdict: WRONG ANSWER

Test 39

Verdict: WRONG ANSWER

Test 40

Verdict: WRONG ANSWER

Test 41

Verdict: WRONG ANSWER

Test 42

Verdict: WRONG ANSWER

Test 43

Verdict: WRONG ANSWER

Test 44

Verdict: WRONG ANSWER

Test 45

Verdict: WRONG ANSWER

Test 46

Verdict: WRONG ANSWER

Test 47

Verdict: WRONG ANSWER

Test 48

Verdict: WRONG ANSWER

Test 49

Verdict: WRONG ANSWER

Test 50

Verdict: WRONG ANSWER

Test 51

Verdict: WRONG ANSWER

Test 52

Verdict: WRONG ANSWER

Test 53

Verdict: WRONG ANSWER

Test 54

Verdict: WRONG ANSWER

Test 55

Verdict: WRONG ANSWER

Test 56

Verdict: WRONG ANSWER

Test 57

Verdict: WRONG ANSWER

Test 58

Verdict: WRONG ANSWER

Test 59

Verdict: WRONG ANSWER

Test 60

Verdict: WRONG ANSWER

Test 61

Verdict: WRONG ANSWER

Test 62

Verdict: WRONG ANSWER

Test 63

Verdict: WRONG ANSWER

Test 64

Verdict: WRONG ANSWER

Test 65

Verdict: WRONG ANSWER

Test 66

Verdict: WRONG ANSWER

Test 67

Verdict: WRONG ANSWER

Test 68

Verdict: WRONG ANSWER

Test 69

Verdict: WRONG ANSWER

Test 70

Verdict: WRONG ANSWER

Test 71

Verdict: WRONG ANSWER

Test 72

Verdict: WRONG ANSWER

Test 73

Verdict: WRONG ANSWER

Test 74

Verdict: WRONG ANSWER

Test 75

Verdict: WRONG ANSWER

Test 76

Verdict: WRONG ANSWER

Test 77

Verdict: WRONG ANSWER

Test 78

Verdict: WRONG ANSWER

Test 79

Verdict: WRONG ANSWER

Test 80

Verdict: WRONG ANSWER

Test 81

Verdict: WRONG ANSWER

Test 82

Verdict: WRONG ANSWER

Test 83

Verdict: WRONG ANSWER

Test 84

Verdict: WRONG ANSWER

Test 85

Verdict: WRONG ANSWER

Test 86

Verdict: WRONG ANSWER

Test 87

Verdict: WRONG ANSWER

Test 88

Verdict: WRONG ANSWER

Test 89

Verdict: WRONG ANSWER

Test 90

Verdict: WRONG ANSWER

Test 91

Verdict: WRONG ANSWER