Submission #1255225

#	Time	Username	Problem	Language	Result	Execution time	Memory
1255225		mingga	Financial Report (JOI21_financial)	C++20	100 / 100	358 ms	74432 KiB

Main

// Author: caption_mingle
#include "bits/stdc++.h"

using namespace std;

#define ln "\n"
#define pb push_back
#define fi first
#define se second
#define all(x) (x).begin(), (x).end()
#define sz(x) ((int)(x).size())
#define ll long long
const int mod = 1e9 + 7;
const int inf = 2e9;
const int N = 3e5 + 7;
int n, a[N], d, f[N], rmq[N][20], rmq2[N][20], rig[N], mnd[N];
pair<int, int> b[N];

vector<int> vec;

int get_min(int l, int r) {
	int k = __lg(r - l + 1);
	return min(rmq[l][k], rmq[r - (1 << k) + 1][k]);
}

int get_max(int l, int r) {
	int k = __lg(r - l + 1);
	return max(rmq2[l][k], rmq2[r - (1 << k) + 1][k]);
}

struct SEGTREE {
	int n;
	vector<int> st;
	SEGTREE(int n) : n(n) {
		st.resize(n * 4 + 4, 0);
	}
	void update(int i, int l, int r, int u, int x) {
		if(l > u or r < u) return;
		if(l == r) {
			st[i] = x;
			return;
		}
		int m = (l + r) >> 1;
		update(i * 2, l, m, u, x);
		update(i * 2 + 1, m + 1, r, u, x);
		st[i] = max(st[i * 2], st[i * 2 + 1]);
	}
	int get(int i, int l, int r, int u, int v) {
		if(l > v or r < u) return 0;
		if(u <= l and r <= v) return st[i];
		int m = (l + r) >> 1;
		return max(get(i * 2, l, m, u, v), get(i * 2 + 1, m + 1, r, u, v));
	}
};

vector<int> ev[N];

signed main() {
	cin.tie(0) -> sync_with_stdio(0);
	#define task ""
	if(fopen(task ".INP", "r")) {
		freopen(task ".INP", "r", stdin);
		freopen(task ".OUT", "w", stdout);
	}
	cin >> n >> d;
	for(int i = 1; i <= n; i++) cin >> a[i], vec.pb(a[i]);
	sort(all(vec));
	vec.erase(unique(all(vec)), vec.end());
	for(int i = 1; i <= n; i++) {
		a[i] = upper_bound(all(vec), a[i]) - vec.begin();
		rmq[i][0] = a[i];
	}
	for(int j = 1; (1 << j) <= n; j++) {
		for(int i = 1; i + (1 << j) - 1 <= n; i++) {
			rmq[i][j] = min(rmq[i][j - 1], rmq[i + (1 << (j - 1))][j - 1]);
		}
	}
	for(int i = d; i <= n; i++) {
		mnd[i] = get_min(i - d + 1, i);
		rmq2[i][0] = mnd[i];
	}
	for(int j = 1; (1 << j) <= n; j++) {
		for(int i = 1; i + (1 << j) - 1 <= n; i++) {
			rmq2[i][j] = max(rmq2[i][j - 1], rmq2[i + (1 << (j - 1))][j - 1]);
		}
	}
	for(int i = 1; i <= n; i++) {
		int l = i + d, r = n;
		rig[i] = n;
		while(l <= r) {
			int m = (l + r) >> 1;
			if(get_max(l, m) > a[i]) {
				r = m - 1;
				rig[i] = m;
			} else l = m + 1; 
		}
	}
	SEGTREE st(n);
	int m = sz(vec);
	for(int i = 1; i <= n; i++) {
		ev[a[i]].pb(i);
	}
	for(int i = m; i > 0; i--) {
		for(int id : ev[i]) {
			f[id] = st.get(1, 1, n, id + 1, rig[id]) + 1;
			st.update(1, 1, n, id, f[id]);
		}
	}
	cout << * max_element(f + 1, f + n + 1) << ln;
    cerr << "\nTime: " << clock() * 1000 / CLOCKS_PER_SEC;
}

Compilation message (stderr)

Main.cpp: In function 'int main()':
Main.cpp:62:24: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   62 |                 freopen(task ".INP", "r", stdin);
      |                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
Main.cpp:63:24: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   63 |                 freopen(task ".OUT", "w", stdout);
      |                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 14 / 14
• Subtask 2: 14 / 14
• Subtask 3: 20 / 20
• Subtask 4: 12 / 12
• Subtask 5: 5 / 5
• Subtask 6: 35 / 35