Submission #375846

#	Time	Username	Problem	Language	Result	Execution time	Memory
375846		Kevin_Zhang_TW	Two Dishes (JOI19_dishes)	C++17	10 / 100	194 ms	32108 KiB

dishes

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define pb emplace_back
#define AI(i) begin(i), end(i)
template<class T> bool chmin(T &a, T b) { return b < a && (a = b, true); }
template<class T> bool chmax(T &a, T b) { return a < b && (a = b, true); }
#ifdef KEV
#define DE(args...) kout("[ " + string(#args) + " ] = ", args)
void kout() { cerr << endl; }
template<class T, class ...U> void kout(T a, U ...b) { cerr << a << ' ', kout(b...); }
template<class T> void debug(T l, T r) { while (l != r) cerr << *l << " \n"[next(l) == r], ++l; }
#else
#define DE(...) 0
#define debug(...) 0
#endif 
const int MAX_N = 2010, MAX_M = 1000010;

const ll inf = 1ll << 60;

ll n[2], a[2][MAX_M], tl[2][MAX_M], w[2][MAX_M];

int ch[2][MAX_M];

ll dur[2][MAX_M];

ll dp[MAX_N][MAX_N];

struct bit {
	int n;
	vector<ll> val;
	bit(int _n = 0) : n(_n + 10) {
		val.resize(n);
	}
	void add(int i, ll v) {
		for (;i < n;i += i&-i)
			val[i] += v;
	}
	ll qry(int i) {
		ll res = 0;
		for (;i;i ^= i&-i) 
			res += val[i];
		return res;
	}
};

ll lost[2][MAX_M];

void hoolan() {
	ll res = 0;

	for (int k = 0;k < 2;++k) {
		for (int i = 1;i <= n[k];++i) {
			res += w[k][i];
		}
	}

	int p[2] {1, 1} ;

	while (true) {
		ll cost = inf, sd = -1;
		for (int k = 0;k < 2;++k) {
			if (p[k] > n[k]) continue;
			int ind = p[k];
			if (chmin(cost, lost[k][ p[k] ]))
				sd = k;
		}
		if (sd == -1) break;
		res -= cost;
	
		DE(sd, p[sd], w[sd][p[sd]]);

		lost[sd^1][ ch[sd][ p[sd] ] ] -= w[sd][ p[sd] ];
		++p[sd];
	}

	cout << res << '\n';
}
int32_t main() {
	ios_base::sync_with_stdio(0), cin.tie(0);
	cin >> n[0] >> n[1];

	for (int k = 0;k < 2;++k)
		for (int i = 1;i <= n[k];++i)
			cin >> a[k][i] >> tl[k][i] >> w[k][i];

	for (int k = 0;k < 2;++k) {
		for (int i = 1;i <= n[k];++i)
			dur[k][i] = dur[k][i-1] + a[k][i];
	}

	for (int k = 0;k < 2;++k) {
		for (int i = 1;i <= n[k];++i) {
			ll ct = dur[k][i], more = tl[k][i] - ct;
			if (more < 0) w[k][i] = 0;
			ch[k][i] = upper_bound(dur[k^1], dur[k^1] + n[k^1] + 1, max(0ll, more)) - dur[k^1];
			lost[k^1][ ch[k][i] ] += w[k][i];
		}
	}


	if (max(n[0], n[1]) > 2000) {
		hoolan();
		return 0;
	}

	for (int i = 0;i <= n[0];++i)
		for (int j = 0;j <= n[1];++j)
			dp[i][j] = -inf;
	dp[0][0] = 0;

	for (int i = 0;i <= n[0];++i)
		for (int j = 0;j <= n[1];++j) {
			ll ct = dur[0][i] + dur[1][j];
			chmax(dp[i+1][j], dp[i][j] +
					(ch[0][i+1] > j ? w[0][i+1] : 0));
					//(ct + a[0][i+1] <= tl[0][i+1] ? w[0][i+1] : 0));
			chmax(dp[i][j+1], dp[i][j] +
					(ch[1][j+1] > i ? w[1][j+1] : 0));
					//(ct + a[1][j+1] <= tl[1][j+1] ? w[1][j+1] : 0));
		}

	cout << dp[n[0]][n[1]] << '\n';
}

Compilation message (stderr)

dishes.cpp: In function 'void hoolan()':
dishes.cpp:64:8: warning: unused variable 'ind' [-Wunused-variable]
   64 |    int ind = p[k];
      |        ^~~
dishes.cpp:14:17: warning: statement has no effect [-Wunused-value]
   14 | #define DE(...) 0
      |                 ^
dishes.cpp:71:3: note: in expansion of macro 'DE'
   71 |   DE(sd, p[sd], w[sd][p[sd]]);
      |   ^~
dishes.cpp: In function 'int32_t main()':
dishes.cpp:114:7: warning: unused variable 'ct' [-Wunused-variable]
  114 |    ll ct = dur[0][i] + dur[1][j];
      |       ^~

• Subtask 1: 0 / 5
• Subtask 2: 3 / 3
• Subtask 3: 7 / 7
• Subtask 4: 0 / 39
• Subtask 5: 0 / 11
• Subtask 6: 0 / 9
• Subtask 7: 0 / 17
• Subtask 8: 0 / 9