제출 #1158078

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1158078		InvMOD	메기 농장 (IOI22_fish)	C++20	3 / 100	55 ms	7240 KiB

fish

#include<bits/stdc++.h>

//#define name "InvMOD"
#ifndef name
    #include "fish.h"
#endif // name

using namespace std;

#define fi first
#define se second

//#define int long long
#define sz(v) (int)(v).size()
#define all(v) (v).begin(), (v).end()

using ll = long long;

namespace Subtask1{
    bool ckSub(vector<int>& X){
        for(int i = 0; i < sz(X); i++){
            if(X[i] & 1) return false;
        }
        return true;
    }

    ll process(vector<int>& W){
        ll answer = 0;
        for(int i = 0; i < sz(W); i++){
            answer += W[i];
        }
        return answer;
    }
}

namespace Subtask2{
    bool ckSub(vector<int>& X){
        for(int i = 0; i < sz(X); i++){
            if(X[i] > 1) return false;
        }
        return true;
    }

    ll process(int n, vector<int>& X, vector<int>& Y, vector<int>& W){
        vector<vector<int>> row(2, vector<int>(n + 1));

        for(int i = 0; i < sz(X); i++){
            row[X[i]][Y[i] + 1] = W[i];
        }

        vector<vector<ll>> pref(2, vector<ll>(n + 1));
        for(int i = 0; i < 2; i++){
            for(int j = 1; j <= n; j++){
                pref[i][j] = pref[i][j - 1] + 1ll * row[i][j];
            }
        }

        ll answer = max(pref[0][n], pref[1][n]);
        for(int i = 0; i <= n; i++){
            for(int j = 0; j <= n; j++){
                if(i > j){
                    answer = max(answer, pref[1][i] - pref[1][j]);
                }
                else{
                    answer = max(answer, pref[0][j] - pref[0][i]);
                }
            }
        }
        return answer;
    }
}

ll max_weights(int n, int m, vector<int> X, vector<int> Y, vector<int> W){

    if(Subtask1::ckSub(X)){
        return Subtask1::process(W);
    }

    if(Subtask2::ckSub(X)){
        return Subtask2::process(n, X, Y, W);
    }

    return 0;
}

#ifdef name
    int32_t main(){
        freopen(name".INP", "r", stdin);
        freopen(name".OUT", "w", stdout);

        int n,m; cin >> n >> m;

        vector<int> X(m), Y(m), W(m);

        for(int i = 0; i < m; i++){
            cin >> X[i];
        }
        for(int i = 0; i < m; i++){
            cin >> Y[i];
        }
        for(int i = 0; i < m; i++){
            cin >> W[i];
        }

        cout << max_weights(n, m, X, Y, W) << "\n";

        return 0;
    }
#endif // name

• 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