Submission #528812

#	Time	Username	Problem	Language	Result	Execution time	Memory
528812		fhvirus	Two Dishes (JOI19_dishes)	C++17	10 / 100	138 ms	13184 KiB

dishes

// Knapsack DP is harder than FFT.
#include<bits/stdc++.h>
using namespace std;
typedef long long ll; typedef pair<int,int> pii;
#define ff first
#define ss second
#define pb emplace_back
#define AI(x) begin(x),end(x)
#ifdef OWO
#define debug(args...) SDF(#args, args)
#define OIU(args...) ostream& operator<<(ostream&O,args)
#define LKJ(S,B,E,F) template<class...T>OIU(S<T...>s){O<<B;int c=0;for(auto i:s)O<<(c++?", ":"")<<F;return O<<E;}
LKJ(vector,'[',']',i)LKJ(deque,'[',']',i)LKJ(set,'{','}',i)LKJ(multiset,'{','}',i)LKJ(unordered_set,'{','}',i)LKJ(map,'{','}',i.ff<<':'<<i.ss)LKJ(unordered_map,'{','}',i.ff<<':'<<i.ss)
template<class...T>void SDF(const char* s,T...a){int c=sizeof...(T);if(!c){cerr<<"\033[1;32mvoid\033[0m\n";return;}(cerr<<"\033[1;32m("<<s<<") = (",...,(cerr<<a<<(--c?", ":")\033[0m\n")));}
template<class T,size_t N>OIU(array<T,N>a){return O<<vector<T>(AI(a));}template<class...T>OIU(pair<T...>p){return O<<'('<<p.ff<<','<<p.ss<<')';}template<class...T>OIU(tuple<T...>t){return O<<'(',apply([&O](T...s){int c=0;(...,(O<<(c++?", ":"")<<s));},t),O<<')';}
#else
#define debug(...) ((void)0)
#endif

signed main() {
	ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);

	int N, M; cin >> N >> M;
	vector<int> A(N + 1), P(N + 1);
	vector<int64_t> S(N + 1);
	for (int i = 1; i <= N; ++i)
		cin >> A[i] >> S[i] >> P[i];
	vector<int> B(M + 1), Q(M + 1);
	vector<int64_t> T(M + 1);
	for (int i = 1; i <= M; ++i)
		cin >> B[i] >> T[i] >> Q[i];

	assert(N <= 5000 and M <= 5000);

	vector<int64_t> preA(N + 1, 0);
	for (int i = 1; i <= N; ++i)
		preA[i] = preA[i - 1] + A[i];

	vector<int64_t> preB(M + 1, 0);
	for (int i = 1; i <= M; ++i)
		preB[i] = preB[i - 1] + B[i];

	vector<int64_t> boundP(N + 1, 0);
	for (int i = 1; i <= N; ++i)
		boundP[i] = (upper_bound(AI(preB), S[i] - preA[i]) - begin(preB)) - 1;

	vector<int64_t> boundQ(M + 1, 0);
	for (int i = 1; i <= M; ++i)
		boundQ[i] = (upper_bound(AI(preA), T[i] - preB[i]) - begin(preA)) - 1;

	int64_t ans = 0;
	vector<int64_t> dp_diff(M + 1, 0);
	for (int j = 0; j <= M; ++j)
		if (boundQ[j] >= N)
			ans += Q[j];

	for (int i = 1; i <= N; ++i) {
		if (boundP[i] >= 0) {
			ans += P[i];
			if (boundP[i] < M)
				dp_diff[boundP[i] + 1] -= P[i];
		}

		for (int j = 0; j <= M; ++j) {
			if (i == boundQ[j] + 1)
				dp_diff[j] += Q[j];
			if (dp_diff[j] < 0) {
				if (j < M) dp_diff[j + 1] += dp_diff[j];
				dp_diff[j] = 0;
			}
		}
		debug(ans, dp_diff);
	}

	cout << accumulate(AI(dp_diff), ans) << '\n';

	return 0;
}

• 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