제출 #628423

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
628423		kingmoshe	메기 농장 (IOI22_fish)	C++17	3 / 100	85 ms	12880 KiB

fish

#include "fish.h"

#include <vector>

bool is_even(std::vector<int> X) {
    for (int i = 0; i < X.size(); i++) {
        if (X[i] % 2 == 1) {
            return false;
        }
    }
    return true;
}

bool is_only_start(std::vector<int> X) {
    for (int i = 0; i < X.size(); i++) {
        if (X[i] > 1) {
            return false;
        }
    }
    return true;
}

long long solve_only_start(int N, int M, std::vector<int> X, std::vector<int> Y,
    std::vector<int> W) {
    if (N == 2) {
        long long res0 = 0;
        long long res1 = 0;
        for (int i = 0; i < M; i++) {
            if (X[i] == 0) {
                res0 += W[i];
            }
            else {
                res1 += W[i];
            }
        }
        if (res0 > res1) {
            return res0;
        }
        return res1;
    }
    long long cur_res = 0;
    for (int i = 0; i < M; i++) {
        if (X[i] == 1) {
            cur_res += W[i];
        }
    }
    std::vector<int> values(N);
    for (int i = 0; i < M; i++) {
        if (X[i] == 0) {
            values[i] += W[i];
        }
        else {
            values[i] -= W[i];
        }
    }
    for (int i = 1; i < N; i++) {
        values[i] += values[i - 1];
    }
    long long best_res = cur_res;
    for (int i = 0; i < N; i++) {
        if (values[i] + cur_res > best_res) {
            best_res = values[i] + cur_res;
        }
    }
    return best_res;
}

long long max_weights(int N, int M, std::vector<int> X, std::vector<int> Y,
    std::vector<int> W) {
    if (is_even(X)) {
        long long res = 0;
        for (int i = 0; i < W.size(); i++) {
            res += W[i];
        }
        return res;
    }
    else if (is_only_start(X)) {
        return solve_only_start(N, M, X, Y, W);
    }
    return 0;
}


/*int main() {

}*/

컴파일 시 표준 에러 (stderr) 메시지

fish.cpp: In function 'bool is_even(std::vector<int>)':
fish.cpp:6:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    6 |     for (int i = 0; i < X.size(); i++) {
      |                     ~~^~~~~~~~~~
fish.cpp: In function 'bool is_only_start(std::vector<int>)':
fish.cpp:15:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   15 |     for (int i = 0; i < X.size(); i++) {
      |                     ~~^~~~~~~~~~
fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:72:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   72 |         for (int i = 0; i < W.size(); i++) {
      |                         ~~^~~~~~~~~~

• Subtask 1: 3 / 3
• Subtask 2: 0 / 6
• Subtask 3: 0 / 9
• Subtask 4: 0 / 14
• Subtask 5: 0 / 21
• Subtask 6: 0 / 17
• Subtask 7: 0 / 14
• Subtask 8: 0 / 16