Submission #702966

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
702966	2023-02-25T11:33:38 Z	n1k	Catfish Farm (IOI22_fish)	C++17	Compilation error	0 ms	0 KB

fish

#include <bits/stdc++.h>

#define ll long long
#define vt vector
#define pb push_back
#define ar array
#define all(x) (x).begin(), (x).end()
#define sz(x) (x).size()

using namespace std;

/*
 ## TAKE IT EASY ##
 1. simplify
 2. add new elements
 3. brute force solution
 4. optimize
 5. start implementing
*/

// --- templates ---

// --- code ---

const int N = 300 + 10, FLOW = 3;


ll dp[N][FLOW][N];

ll max_weights(int n, int m, vt<int> X, vt<int> Y, vt<int> W){
	memset(dp, 0, sizeof dp);

	int h = n + 5;

	vt<vt<ll>> a(n + 5, vt<ll>(h));

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

	for(int i = 0; i < n; i++){
		for(int j = 1; j < h; j++){
			a[i][j] += a[i][j - 1];
		}
	}

	for(int i = 0; i < n; i++){
		for(int flow = 0; flow < FLOW; flow++){
			for(int h1 = 0; h1 < h; h1++){
				for(int h2 = 0; h2 < (flow == 0 ? h1 + 1 : h); h2++){

					int nflow = h1 <= h2;
					if(i == 0){
						dp[i][nflow][h2] = max(dp[i][nflow][h2], a[i + 1][h2]);
					}else{
						if(flow == 0){
							dp[i][nflow][h2] = max(dp[i][nflow][h2], dp[i - 1][flow][h1] + a[i + 1][h2] - a[i][h2]);
						}else if(flow == 1){
							dp[i][nflow][h2] = max(dp[i][nflow][h2], dp[i - 1][flow][h1] + a[i + 1][h2] + max((ll)0, (a[i - 1][h2] - a[i - 1][h1])) - min(a[i][h1], a[i][h1]));
						}else{
							nflow = 1;
							dp[i][nflow][h2] = max(dp[i][nflow][h2], dp[i - 1][flow][h1] + a[i + 1][h2] + min((ll)0, (a[i - 1][h2] - a[i - 1][h1])));
						}
					}
				}
				if(i)
					dp[i][2][h1] = max(dp[i - 1][0][h1], dp[i - 1][1][h1]);
			}
		}
	}

	ll ans = 0;

	for(int flow = 0; flow < FLOW; flow++){
		for(int h1 = 0; h1 < h; h1++){
			ans = max(ans, dp[n - 1][flow][h1]);
		}
	}

	return ans;
}

int main(){
	int n, m;
	cin >> n >> m;
	vt<int> x(m), y(m), w(m);
	for(int i = 0; i < m; i++){
		cin >> x[i] >> y[i] >> w[i];
	}
	cout << max_weights(n, m, x, y, w) << endl;

	return 0;
}

Compilation message

/usr/bin/ld: /tmp/ccmwHoZe.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccTMQopf.o:fish.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status