Submission #975784

#	Time	Username	Problem	Language	Result	Execution time	Memory
975784		ALeonidou	메기 농장 (IOI22_fish)	C++17	0 / 100	26 ms	7516 KiB

fish

#include "fish.h"
#include <bits/stdc++.h>
using namespace std;

#define ll int
#define sz(x) (ll)x.size()
#define F first
#define S second
#define endl "\n"
#define pb push_back

typedef vector <ll> vi;
typedef vector <long long> vl;
typedef pair <ll,ll> ii;
typedef vector <ii> vii;

#define dbg(x) cout<<#x<<": "<<x<<endl;
#define dbg2(x,y) cout<<#x<<": "<<x<<" "<<#y<<": "<<y<<endl;
#define dbg3(x,y,z) cout<<#x<<": "<<x<<" "<<#y<<": "<<y<<" "<<#z<<": "<<z<<endl;

void printVct(vl &v){
    for(ll i =0; i<sz(v); i++){
        cout<<v[i]<<" ";
    }
    cout<<endl;
}

long long getSum(ll l, ll r, const vl &s){
    if (l == 0) return s[r];
    return s[r] - s[l-1];
}

#define N 100001
vl arr;
ll n, m;
ll dp[N+1];
long long solve(ll i){  //wall already present at position c
    if (i >= n) return 0;
    if (i == n-2) return arr.back();
    if (dp[i] != -1) return dp[i];

    long long ans =0;

    //i+1
    if (i < n-1){
        ans = max(ans, solve(i+1));
    }

    //i+2
    if (i < n-2){
        ans = max(ans, solve(i+2) + arr[i+1]);
    }

    //i+3
    if (i < n-3){
        ans = max(ans, solve(i+3) + arr[i+1] + arr[i+2]);
    }
        
    dp[i] = ans;
    return dp[i];
}

void init_dp(){
    for (ll i =0; i<=n; i++){
        dp[i] = -1;
    }   
}

long long max_weights(int NN, int M, vi X, vi Y, vi w){
    n = NN, m = M;

    //create arr
    arr.assign(n, 0);
    for (ll i =0; i<m; i++){
        ll x = X[i];
        arr[x] = w[i];
    }

    // cout<<"arr: ";
    // printVct(arr);

    long long ans = 0;
    init_dp();
    ans = solve(0);

    for (ll i = 1; i<=min(2, n-1); i++){
        init_dp();
        ans = max(ans, solve(i) + arr[i-1]);
    }

    return ans;
}

/*
7 7
0 0 2
1 0 1
2 0 0
3 0 5
4 0 3
5 0 3
6 0 4


6 4
0 2 5
2 1 4
4 4 1
6 3 3

6 6
0 2 5
1 1 4
0 4 1
0 3 3
1 2 4
1 6 2

2 3
0 1 2
0 0 4
1 1 3


5 4
0 2 5
1 1 2
4 4 1
3 3 3

*/

• Subtask 1: 0 / 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