제출 #1120010

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1120010		Nonoze	메기 농장 (IOI22_fish)	C++17	0 / 100	1020 ms	219672 KiB

fish

#include "fish.h"

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

#ifndef _IN_LOCAL
	#define dbg(...)
#endif

#define endl '\n'
#define endlfl '\n' << flush
#define quit(x) return (void)(cout << x << endl)

template<typename T> void read(T& x) { cin >> x;}
template<typename T1, typename T2> void read(pair<T1, T2>& p) { read(p.first), read(p.second);}
template<typename T> void read(vector<T>& v) { for (auto& x : v) read(x); }
template<typename T1, typename T2> void read(T1& x, T2& y) { read(x), read(y); }
template<typename T1, typename T2, typename T3> void read(T1& x, T2& y, T3& z) { read(x), read(y), read(z); }
template<typename T1, typename T2, typename T3, typename T4> void read(T1& x, T2& y, T3& z, T4& zz) { read(x), read(y), read(z), read(zz); }
template<typename T> void print(vector<T>& v) { for (auto& x : v) cout << x << ' '; cout << endl; }

#define sz(x) (int)(x.size())
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define pb push_back
#define mp(a, b) make_pair(a, b)
#define fi first
#define se second
#define cmin(a, b) a = min(a, b)
#define cmax(a, b) a = max(a, b)
#define YES cout << "YES" << endl
#define NO cout << "NO" << endl
#define QYES quit("YES")
#define QNO quit("NO")

#define int long long
#define double long double
const int inf = numeric_limits<int>::max() / 4;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const int MOD = 1e9+7, LOG=20;



int n;
vector<vector<pair<int, int>>> pref, is;
map<vector<int>, int> memo;

int calc(int x, int l, int r) {
	if (l>r || x>=n || x<0) return 0;
	auto lb=upper_bound(all(pref[x]), mp(r, -1LL)); if (lb==pref[x].begin()) return 0;
	lb--;
	if (l==0) return lb->se;
	int ans=lb->se;
	lb=lower_bound(all(pref[x]), mp(l, -1LL)); if (lb==pref[x].begin()) return ans;
	lb--;
	return ans-lb->se;
}



int dp(int i, int j, int nb , bool before, bool taking) {
	if (nb>=3 || i>=n) return 0;
	if (memo.count({i, j, nb, before, taking})) return memo[{i, j, nb, before, taking}];
	int ans=0;
	if (taking) {
		if (nb==2) {
			cmax(ans, dp(i+1, j, 0, 0, 0)+calc(i-1, 0, is[i][j].fi-1)+calc(i+1, 0, is[i][j].fi-1));
			if (j+1<sz(is[i])) cmax(ans, dp(i, j+1, 2, 0, 1));
		} else if (nb==1) {
			cmax(ans, dp(i+1, j, 0, 0, 0)+calc(i+1, 0, is[i][j].fi-1));
			if (j+1<sz(is[i])) cmax(ans, dp(i, j+1, 1, 0, 1));
		} else {
			cmax(ans, dp(i+1, j, 0, 0, 0)+calc(i+1, 0, is[i][j].fi-1));
			if (before && j) cmax(ans, dp(i, j-1, 0, 1, 1)-is[i][j-1].se);
			else if (!before && j+1<sz(is[i])) cmax(ans, dp(i, j+1, 0, 0, 1));
		}
	} else{
		cmax(ans, dp(i+1, j, nb+1, 0, 0));
		if (nb==0) {
			int empl=lower_bound(all(is[i]), mp(is[i-1][j].fi, -1LL))-is[i].begin(), clc=pref[i][empl].se-is[i][empl].se;
			cmax(ans, dp(i, empl, 0, 0, 1)-clc);
			cmax(ans, dp(i, empl, 0, 1, 1)-clc);
		} else if (nb==1) {
			cmax(ans, dp(i, 0, nb, 0, 1));
		} else {
			cmax(ans, dp(i, 0, nb, 0, 1));
		}
	}
	return memo[{i, j, nb, before, taking}]=ans;
}


int max_weights(signed N, signed m, vector<signed> x, vector<signed> y, vector<signed> w) {
	n=N;
	pref.resize(n), is.resize(n);
	for (int i=0; i<m; i++) {
		is[x[i]].pb({y[i], w[i]});
	}
	for (int i=0; i<n; i++) {
		sort(all(is[i]));
		is[i].pb({n, 0});
		int sm=0;
		for (auto [j, x]: is[i]) {
			sm+=x;
			pref[i].pb({j, sm});
		}
	}
	return dp(0, 0, 1, 0, 0);
}

• 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