Submission #496671

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
496671	2021-12-21T19:17:23 Z	Ziel	Meteors (POI11_met)	C++17	100 / 100	1918 ms	53132 KB

/**
 * LES GREATEABLES BRO TEAM
**/

#include <bits/stdc++.h>

using namespace std;

using ll = long long;
#define sz(x) (int)x.size()
const bool FLAG = false;
void setIO(const string &f = "");

string to_string(const string s) {
	return '"' + s + '"';
}
string to_string(const char c) {
 	return char(39) + string(1, c) + char(39);
}
string to_string(const char* s) {
 	return to_string(string(s));
}
string to_string(bool f) {
	return (f ? "true" : "false");
}
template<class A, class B>
string to_string(const pair<A, B> x) {
	return "(" + to_string(x.first) + ", " + to_string(x.second) + ")";
}
template<class A, class B, class C>
string to_string(const tuple<A, B, C> x) {
	return "(" + to_string(get<0>(x)) + ", " + to_string(get<1>(x)) + ", " + to_string(get<2>(x)) + ")";
}
template<class A, class B, class C, class D>
string to_string(const tuple<A, B, C, D> x) {
	return "(" + to_string(get<0>(x)) + ", " + to_string(get<1>(x)) + ", " + to_string(get<2>(x)) + ", " + to_string(get<3>(x)) + ")";
}
template<class T>
string to_string(const T v) {
	string res = "{"; bool f = true;
	for (auto x: v)
		res += (f ? to_string(x) : ", " + to_string(x)), f = false;
	return res + "}";
}
void debug_args() { cerr << "]\n"; }
template<class H, class... T>
void debug_args(H x, T... y) {
	cerr << to_string(x);
	if (sizeof... (y))
	cerr << ", ";
	debug_args(y...);
}

#ifdef LOCAL
#define debug(...) cerr << "[" << #__VA_ARGS__ << "]: [", debug_args(__VA_ARGS__);
#else
#define debug(...) 47
#endif

#define int ll

const int N = 3e6 + 13;
int n, m;
ll bit[N];

void upd(int x, ll v) {
	for (; x <= m; x |= (x + 1))
		bit[x] += v;
}
ll get(int x) {
	ll res = 0;
	for (; x > 0; x = (x & (x + 1)) - 1)
		res += bit[x];
	return res;
}
void clear() {
	for (int x = 1; x <= m; x++)
		bit[x] = 0;
}

void solve() {
	cin >> n >> m;
	vector<int> o(m + 1);
	vector<vector<int>> occ(n + 1);
	for (int i = 1; i <= m; i++) {
		cin >> o[i];
		occ[o[i]].push_back(i);
	}
	vector<int> p(n + 1);
	for (int i = 1; i <= n; i++)
		cin >> p[i];
	int q;
	cin >> q;
	vector<int> l(q + 1), r(q + 1), a(q + 1);
	for (int i = 1; i <= q; i++) {
	    cin >> l[i] >> r[i] >> a[i];
	}

	vector<int> left(n + 1, 1), right(n + 1, q), res(n + 1, -1);
	while (true) {
		clear();
		vector<vector<int>> check(q + 1);
		bool over = true;
		for (int i = 1; i <= n; i++) {
			if (left[i] <= right[i]) {
				int mid = (left[i] + right[i]) / 2;
				check[mid].push_back(i);
				over = false;
			}
		}
		if (over)
			break;
		for (int i = 1; i <= q; i++) {
			if (l[i] <= r[i]) {
				upd(l[i], a[i]);
				upd(r[i] + 1, -a[i]);
			} else {
				upd(l[i], a[i]);
				upd(m + 1, -a[i]);
				upd(1, a[i]);
				upd(r[i] + 1, -a[i]);
			}

			for (int x: check[i]) {
				ll cur = 0;
				for (int pos: occ[x]) {
					cur += get(pos);
					if (cur >= p[x])
						break;
				}
				if (cur >= p[x]) {
					right[x] = i - 1;
					res[x] = i;
				} else
					left[x] = i + 1;
			}
		}
	}
	for (int i = 1; i <= n; i++) {
		if (res[i] == -1)
			cout << "NIE\n";
		else
			cout << res[i] << '\n';
	}
}

signed main() {
    setIO();
    
  	solve();

    return 0;
}

void setIO(const string &f) {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    if (fopen((f + ".in").c_str(), "r")) {
        freopen((f + ".in").c_str(), "r", stdin);
        freopen((f + ".out").c_str(), "w", stdout);
    }
}

Compilation message

met.cpp: In function 'void setIO(const string&)':
met.cpp:159:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  159 |         freopen((f + ".in").c_str(), "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
met.cpp:160:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  160 |         freopen((f + ".out").c_str(), "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	332 KB	Output is correct
2	Correct	1 ms	332 KB	Output is correct
3	Correct	1 ms	332 KB	Output is correct

Subtask #2
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	332 KB	Output is correct
2	Correct	2 ms	332 KB	Output is correct
3	Correct	2 ms	460 KB	Output is correct

Subtask #3
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	52 ms	4444 KB	Output is correct
2	Correct	97 ms	6176 KB	Output is correct
3	Correct	80 ms	5276 KB	Output is correct

Subtask #4
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	76 ms	4832 KB	Output is correct
2	Correct	79 ms	4872 KB	Output is correct
3	Correct	90 ms	6156 KB	Output is correct
4	Correct	30 ms	3196 KB	Output is correct

Subtask #5
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	66 ms	4508 KB	Output is correct
2	Correct	77 ms	6120 KB	Output is correct
3	Correct	36 ms	2752 KB	Output is correct
4	Correct	79 ms	5532 KB	Output is correct

Subtask #6
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	77 ms	4004 KB	Output is correct
2	Correct	86 ms	4940 KB	Output is correct
3	Correct	58 ms	4224 KB	Output is correct
4	Correct	109 ms	6328 KB	Output is correct

Subtask #7
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1112 ms	32320 KB	Output is correct
2	Correct	695 ms	21552 KB	Output is correct
3	Correct	191 ms	14432 KB	Output is correct
4	Correct	1704 ms	49604 KB	Output is correct

Subtask #8
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1108 ms	31252 KB	Output is correct
2	Correct	765 ms	21572 KB	Output is correct
3	Correct	171 ms	13172 KB	Output is correct
4	Correct	1918 ms	53132 KB	Output is correct