Submission #496617

# Submission time Handle Problem Language Result Execution time Memory
496617 2021-12-21T16:38:30 Z Ziel Meteors (POI11_met) C++17
74 / 100
6000 ms 24332 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 = 3e5 + 13, K = 100;
vector<ll> bit(N), prev_bit(N);
int n, m;

void upd(int x, ll v) {
	for (; x <= m + 1; 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 prev_upd(int x, ll v) {
	for (; x <= m + 1; x |= (x + 1))
		prev_bit[x] += v;
}
ll prev_get(int x) {
	ll res = 0;
	for (; x > 0; x = (x & (x + 1)) - 1)
		res += prev_bit[x];
	return res;
}

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;
    int cnt_rep = 0;
    vector<int> l(q + 1), r(q + 1), a(q + 1), ans(n + 1, -1);
    for (int rep = 1; rep <= q; rep++) {
		cin >> l[rep] >> r[rep] >> a[rep];
		if (l[rep] <= r[rep]) {
			upd(l[rep], a[rep]);
			upd(r[rep] + 1, -a[rep]);
		} else {
			upd(l[rep], a[rep]);
			upd(m + 1, -a[rep]);
			upd(1, a[rep]);
			upd(r[rep] + 1, -a[rep]);
		}


    	cnt_rep++;
		if (cnt_rep >= K) {
			vector<int> cur_p(n + 1);
			for (int i = 1; i <= m; i++)
				cur_p[o[i]] += get(i);
			set<int> temp;
			for (int i = 1; i <= n; i++) {
				if (ans[i] == -1 && cur_p[i] >= p[i])
					temp.insert(i);
			}

			for (int i = rep - K + 1; i <= rep; i++) {
				if (l[i] <= r[i]) {
					prev_upd(l[i], a[i]);
					prev_upd(r[i] + 1, -a[i]);
				} else {
					prev_upd(l[i], a[i]);
					prev_upd(m + 1, -a[i]);
					prev_upd(1, a[i]);
					prev_upd(r[i] + 1, -a[i]);
				}
				
				if (sz(temp)) {
					vector<int> del;
					for (int x: temp) {
						int cur = 0;
						for (int y: occ[x]) {
							cur += prev_get(y);
						}
						if (cur >= p[x]) {
							ans[x] = i;
							del.push_back(x);
						}
					}
					for (int x: del)
						temp.erase(x);
				} else {
					prev_bit = bit;
					break;
				}
			}
			cnt_rep = 0;
		}
    }

	{
	    vector<int> cur_p(n + 1);
		for (int i = 1; i <= m; i++)
			cur_p[o[i]] += get(i);
		set<int> temp;
		for (int i = 1; i <= n; i++) {
			if (ans[i] == -1 && cur_p[i] >= p[i])
				temp.insert(i);
		}

		for (int i = q - cnt_rep + 1; i <= q; i++) {
			if (l[i] <= r[i]) {
				prev_upd(l[i], a[i]);
				prev_upd(r[i] + 1, -a[i]);
			} else {
				prev_upd(l[i], a[i]);
				prev_upd(m + 1, -a[i]);
				prev_upd(1, a[i]);
				prev_upd(r[i] + 1, -a[i]);
			}
		
			if (sz(temp)) {
				vector<int> del;
				for (int x: temp) {
					int cur = 0;
					for (int y: occ[x]) {
						cur += prev_get(y);
					}
					if (cur >= p[x]) {
						ans[x] = i;
						del.push_back(x);
					}
				}
				for (int x: del)
					temp.erase(x);
			}
		}
	}

	for (int i = 1; i <= n; i++) {
		if (ans[i] == -1)
			cout << "NIE\n";
		else
			cout << ans[i] << '\n';
	}
}

signed main() {
    setIO();
    
    int tt = 1;
    if (FLAG) {
    	cin >> tt;
    }
    while (tt--) {
    	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:227:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  227 |         freopen((f + ".in").c_str(), "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
met.cpp:228:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  228 |         freopen((f + ".out").c_str(), "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 4 ms 5068 KB Output is correct
2 Correct 3 ms 5068 KB Output is correct
3 Correct 4 ms 5068 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 4 ms 5068 KB Output is correct
2 Correct 4 ms 5104 KB Output is correct
3 Correct 5 ms 5112 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 337 ms 7404 KB Output is correct
2 Correct 392 ms 8632 KB Output is correct
3 Correct 371 ms 7960 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 372 ms 7796 KB Output is correct
2 Correct 372 ms 7720 KB Output is correct
3 Correct 519 ms 9292 KB Output is correct
4 Correct 61 ms 7112 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 322 ms 7504 KB Output is correct
2 Correct 371 ms 8544 KB Output is correct
3 Correct 67 ms 6220 KB Output is correct
4 Correct 430 ms 8164 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 331 ms 7112 KB Output is correct
2 Correct 433 ms 7668 KB Output is correct
3 Correct 325 ms 7260 KB Output is correct
4 Correct 404 ms 8820 KB Output is correct
# Verdict Execution time Memory Grader output
1 Execution timed out 6017 ms 24332 KB Time limit exceeded
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Execution timed out 6010 ms 23472 KB Time limit exceeded
2 Halted 0 ms 0 KB -