Niitty

#include <iostream>
#include <stdlib.h>
#include <algorithm>    // std::sort
 
unsigned int n, m;
 
unsigned int *stick_lengths;
 
 
inline unsigned int min(unsigned int a, unsigned int b) {
    return (a < b ? a : b);
}
 
inline unsigned int max(unsigned int a, unsigned int b) {
    return (a > b ? a : b);
}
 
void n_is_one_sol() {
    for (unsigned int k = 1; k <= m; k++) {
        if (stick_lengths[0] % k == 0) {
            std::cout << "0 ";
        } else {
            std::cout << "1 ";
        }
    }
}
 
void p1_sol() {
    /* Osatehtävä 1 (7 pistettä)
 
        1 <= n <= 5
        1 <= m <= 10
        1 <= a_i <= 10 */
    // brute force? numbers are really small
    throw;
    for (unsigned int k = 1; k <= m; k++) {
        
    }
}
 
// WORKS
void p2_sol() {
    /* Osatehtävä 2 (8 pistettä)
 
        1 <= n <= 1000
        1 <= m <= 2000
        1 <= a_i <= 3  */
    unsigned int one_count = 0;
    unsigned int two_count = 0;
    unsigned int three_count = 0;
 
    unsigned int *solutions = (unsigned int*) malloc(m * sizeof(unsigned int));
 
    for (unsigned int i = 0; i < n; i++ ) {
        switch (stick_lengths[i]) {
            case 1:
                one_count++;
                break;
            case 2:
                two_count++;
                break;
            case 3:
                three_count++;
                break;
        }
    }
 
    unsigned int cuts_left = m;
    unsigned int i = 0;
    while (three_count > 0 && cuts_left > 0) {
        three_count--;
        one_count++;
        two_count++;
        if (three_count > 0) {
            solutions[i] = 2;
        } else {
            solutions[i] = 1;
        }
        cuts_left--;
        i++;
    }
    while (two_count > 0 && cuts_left > 0) {
        two_count--;
        one_count += 2;
        if (two_count > 0) {
            solutions[i] = 1;
        } else {
            solutions[i] = 0;
        }
        cuts_left--;
        i++;
    }
 
    std::cout << solutions[0];
    for (unsigned int j = 1; j < m; j++) {
        std::cout << ' ' << solutions[j];
    }
}
 
// WORKS
void p3_sol() {
    /* Osatehtävä 3 (12 pistettä)
 
        1 <= n <= 1000
        1 <= m <= 2
        1 <= a_i <= 1000 */
    unsigned int max_stick = stick_lengths[n-1];
    unsigned int best_sol_p1 = -1;
    for (unsigned int new_stick1_size = 1; new_stick1_size < max_stick; new_stick1_size++) {
        unsigned int new_stick2_size = max_stick - new_stick1_size;
        unsigned int min_stick = min(new_stick1_size, min(new_stick2_size, stick_lengths[0]));
        unsigned int max_stick = max(new_stick1_size, max(new_stick2_size, stick_lengths[n-2]));
 
        best_sol_p1 = min(best_sol_p1, max_stick - min_stick);
    }
    std::cout << best_sol_p1;
    
    if (m == 2) {
        // i don't care if the code is a little slow
        // nums are so small anyway...
        unsigned int best_sol_p2 = -1;
        
        unsigned int max_stick = stick_lengths[n-1];
        
        // if we cut the last stick into 3 parts
        for (unsigned int new_stick1_size = 1; new_stick1_size < max_stick - 1; new_stick1_size++) {
            for (unsigned int new_stick2_size = 1; new_stick2_size < max_stick - new_stick1_size; new_stick2_size++) {
                unsigned int new_stick3_size = max_stick - new_stick1_size - new_stick2_size;
                unsigned int min_stick = min(new_stick1_size, min(new_stick2_size, min(new_stick3_size, stick_lengths[0])));
                unsigned int max_stick = max(new_stick1_size, max(new_stick2_size, max(new_stick3_size, stick_lengths[n-2])));
 
                best_sol_p2 = min(best_sol_p2, max_stick - min_stick);
            }
        }
        // if we cut the last stick into 2 parts and the second to last stick into 2 parts
        // new_stick1, new_stick2 are from last stick
        // new_stick3, new_stick4 are from second to last stick
        for (unsigned int new_stick1_size = 1; new_stick1_size < max_stick; new_stick1_size++) {
            for (unsigned int new_stick3_size = 1; new_stick3_size < stick_lengths[n-2]; new_stick3_size++) {
                unsigned int new_stick2_size = max_stick - new_stick1_size;
                unsigned int new_stick4_size = stick_lengths[n-2] - new_stick3_size;
                unsigned int smallest_stick = (n >= 3 ? stick_lengths[0] : new_stick1_size);
                unsigned int largest_stick = (n >= 3 ? stick_lengths[n-3] : new_stick1_size);
                unsigned int min_stick = min(new_stick1_size, min(new_stick2_size, min(new_stick3_size, min(new_stick4_size, smallest_stick))));
                unsigned int max_stick = max(new_stick1_size, max(new_stick2_size, max(new_stick3_size, max(new_stick4_size, largest_stick))));
 
                best_sol_p2 = min(best_sol_p2, max_stick - min_stick);
            }
        }
 
        std::cout << ' ' << best_sol_p2;
    }
}
 
int main() {
    std::cin >> n >> m;
    stick_lengths = (unsigned int*) malloc((n) * sizeof(unsigned int));
    for (unsigned int i = 0; i < n; i++) {
        std::cin >> stick_lengths[i];
    }
    std::sort(stick_lengths, &stick_lengths[n]);
    /*for (int i = 0; i < n; i++) {
        std::cout << stick_lengths[i] << ' ';
    }*/
    
    if (n == 1) {
        n_is_one_sol();
    } else if (m <= 2 && n <= 1000 && stick_lengths[n-1] <= 1000) {
        p3_sol();
    } else if (n <= 5 && m <= 10 && stick_lengths[n-1] <= 10) {
        p1_sol();
    } else {
        // p4, p5, p6 don't get points anyway :)
        p2_sol();
    }
    std::cout << std::endl;
 
    return 0;
}

Test 1

Group: 1, 2, 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 2

Group: 1, 2, 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 3

Group: 1, 2, 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 4

Group: 1, 2, 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 5

Group: 1, 2, 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 6

Group: 2, 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 7

Group: 2, 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 8

Group: 2, 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 9

Group: 2, 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 10

Group: 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 11

Group: 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 12

Group: 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 13

Group: 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 14

Group: 4, 5, 6

Verdict: WRONG ANSWER

Test 15

Group: 4, 5, 6

Verdict: WRONG ANSWER

Test 16

Group: 4, 5, 6

Verdict: WRONG ANSWER

Test 17

Group: 4, 5, 6

Verdict: WRONG ANSWER

Test 18

Group: 5, 6

Verdict: WRONG ANSWER

Test 19

Group: 5, 6

Verdict: WRONG ANSWER

Test 20

Group: 5, 6

Verdict: WRONG ANSWER

Test 21

Group: 5, 6

Verdict: WRONG ANSWER

Test 22

Group: 6

Verdict: WRONG ANSWER

Test 23

Group: 6

Verdict: WRONG ANSWER

Test 24

Group: 6

Verdict: WRONG ANSWER

Test 25

Group: 6

Verdict: WRONG ANSWER