Submission #528820

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
528820	2022-02-21T14:06:45 Z	fhvirus	Two Dishes (JOI19_dishes)	C++17	10 / 100	146 ms	14040 KB

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);
	vector<vector<int64_t>> out(N + 1);
	for (int j = 0; j <= M; ++j)
		if (boundQ[j] >= N)
			ans += Q[j];
		else if (boundQ[j] >= 0)
			out[boundQ[j] + 1].emplace_back(j);

	set<int> not_zero;
	vector<int> modified;
	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];
				not_zero.emplace(boundP[i] + 1);
				modified.emplace_back(boundP[i] + 1);
			}
		}

		for (const int& j: out[i]) {
			dp_diff[j] += Q[j];
			not_zero.emplace(j);
			modified.emplace_back(j);
		}

		sort(AI(modified));
		for (const int& j: modified) {
			auto it = not_zero.lower_bound(j);
			auto ti = it;
			int64_t lst = 0;
			while (it != end(not_zero)) {
				dp_diff[*it] += lst;
				if (dp_diff[*it] >= 0) break;
				lst = dp_diff[*it];
				dp_diff[*it] = 0;
				++it;
			}
			not_zero.erase(ti, it);
		}
		modified.clear();
	}

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

	return 0;
}

Subtask #1
0 / 5.0

#	Verdict	Execution time	Memory	Grader output
1	Runtime error	146 ms	14040 KB	Execution killed with signal 6
2	Halted	0 ms	0 KB	-

Subtask #2
3.0 / 3.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	332 KB	Output is correct
2	Correct	0 ms	204 KB	Output is correct
3	Correct	1 ms	204 KB	Output is correct
4	Correct	1 ms	204 KB	Output is correct
5	Correct	1 ms	316 KB	Output is correct
6	Correct	0 ms	204 KB	Output is correct
7	Correct	1 ms	204 KB	Output is correct
8	Correct	0 ms	204 KB	Output is correct
9	Correct	0 ms	204 KB	Output is correct
10	Correct	1 ms	204 KB	Output is correct
11	Correct	0 ms	204 KB	Output is correct
12	Correct	0 ms	204 KB	Output is correct
13	Correct	0 ms	204 KB	Output is correct
14	Correct	0 ms	204 KB	Output is correct
15	Correct	0 ms	204 KB	Output is correct
16	Correct	0 ms	204 KB	Output is correct

Subtask #3
7.0 / 7.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	332 KB	Output is correct
2	Correct	0 ms	204 KB	Output is correct
3	Correct	1 ms	204 KB	Output is correct
4	Correct	1 ms	204 KB	Output is correct
5	Correct	1 ms	316 KB	Output is correct
6	Correct	0 ms	204 KB	Output is correct
7	Correct	1 ms	204 KB	Output is correct
8	Correct	0 ms	204 KB	Output is correct
9	Correct	0 ms	204 KB	Output is correct
10	Correct	1 ms	204 KB	Output is correct
11	Correct	0 ms	204 KB	Output is correct
12	Correct	0 ms	204 KB	Output is correct
13	Correct	0 ms	204 KB	Output is correct
14	Correct	0 ms	204 KB	Output is correct
15	Correct	0 ms	204 KB	Output is correct
16	Correct	0 ms	204 KB	Output is correct
17	Correct	2 ms	588 KB	Output is correct
18	Correct	2 ms	716 KB	Output is correct
19	Correct	3 ms	588 KB	Output is correct
20	Correct	2 ms	576 KB	Output is correct
21	Correct	3 ms	572 KB	Output is correct
22	Correct	2 ms	588 KB	Output is correct
23	Correct	2 ms	588 KB	Output is correct

Subtask #4
0 / 39.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	332 KB	Output is correct
2	Correct	0 ms	204 KB	Output is correct
3	Correct	1 ms	204 KB	Output is correct
4	Correct	1 ms	204 KB	Output is correct
5	Correct	1 ms	316 KB	Output is correct
6	Correct	0 ms	204 KB	Output is correct
7	Correct	1 ms	204 KB	Output is correct
8	Correct	0 ms	204 KB	Output is correct
9	Correct	0 ms	204 KB	Output is correct
10	Correct	1 ms	204 KB	Output is correct
11	Correct	0 ms	204 KB	Output is correct
12	Correct	0 ms	204 KB	Output is correct
13	Correct	0 ms	204 KB	Output is correct
14	Correct	0 ms	204 KB	Output is correct
15	Correct	0 ms	204 KB	Output is correct
16	Correct	0 ms	204 KB	Output is correct
17	Correct	2 ms	588 KB	Output is correct
18	Correct	2 ms	716 KB	Output is correct
19	Correct	3 ms	588 KB	Output is correct
20	Correct	2 ms	576 KB	Output is correct
21	Correct	3 ms	572 KB	Output is correct
22	Correct	2 ms	588 KB	Output is correct
23	Correct	2 ms	588 KB	Output is correct
24	Runtime error	125 ms	13876 KB	Execution killed with signal 6
25	Halted	0 ms	0 KB	-

Subtask #5
0 / 11.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	332 KB	Output is correct
2	Correct	0 ms	204 KB	Output is correct
3	Correct	1 ms	204 KB	Output is correct
4	Correct	1 ms	204 KB	Output is correct
5	Correct	1 ms	316 KB	Output is correct
6	Correct	0 ms	204 KB	Output is correct
7	Correct	1 ms	204 KB	Output is correct
8	Correct	0 ms	204 KB	Output is correct
9	Correct	0 ms	204 KB	Output is correct
10	Correct	1 ms	204 KB	Output is correct
11	Correct	0 ms	204 KB	Output is correct
12	Correct	0 ms	204 KB	Output is correct
13	Correct	0 ms	204 KB	Output is correct
14	Correct	0 ms	204 KB	Output is correct
15	Correct	0 ms	204 KB	Output is correct
16	Correct	0 ms	204 KB	Output is correct
17	Correct	2 ms	588 KB	Output is correct
18	Correct	2 ms	716 KB	Output is correct
19	Correct	3 ms	588 KB	Output is correct
20	Correct	2 ms	576 KB	Output is correct
21	Correct	3 ms	572 KB	Output is correct
22	Correct	2 ms	588 KB	Output is correct
23	Correct	2 ms	588 KB	Output is correct
24	Runtime error	125 ms	13876 KB	Execution killed with signal 6
25	Halted	0 ms	0 KB	-

Subtask #6
0 / 9.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	332 KB	Output is correct
2	Correct	0 ms	204 KB	Output is correct
3	Correct	1 ms	204 KB	Output is correct
4	Correct	1 ms	204 KB	Output is correct
5	Correct	1 ms	316 KB	Output is correct
6	Correct	0 ms	204 KB	Output is correct
7	Correct	1 ms	204 KB	Output is correct
8	Correct	0 ms	204 KB	Output is correct
9	Correct	0 ms	204 KB	Output is correct
10	Correct	1 ms	204 KB	Output is correct
11	Correct	0 ms	204 KB	Output is correct
12	Correct	0 ms	204 KB	Output is correct
13	Correct	0 ms	204 KB	Output is correct
14	Correct	0 ms	204 KB	Output is correct
15	Correct	0 ms	204 KB	Output is correct
16	Correct	0 ms	204 KB	Output is correct
17	Correct	2 ms	588 KB	Output is correct
18	Correct	2 ms	716 KB	Output is correct
19	Correct	3 ms	588 KB	Output is correct
20	Correct	2 ms	576 KB	Output is correct
21	Correct	3 ms	572 KB	Output is correct
22	Correct	2 ms	588 KB	Output is correct
23	Correct	2 ms	588 KB	Output is correct
24	Runtime error	125 ms	13876 KB	Execution killed with signal 6
25	Halted	0 ms	0 KB	-

Subtask #7
0 / 17.0

#	Verdict	Execution time	Memory	Grader output
1	Runtime error	146 ms	14040 KB	Execution killed with signal 6
2	Halted	0 ms	0 KB	-

Subtask #8
0 / 9.0

#	Verdict	Execution time	Memory	Grader output
1	Runtime error	146 ms	14040 KB	Execution killed with signal 6
2	Halted	0 ms	0 KB	-

Submission #528820

Compilation message