#1018468 (C++23) No.2930 Larger Mex

提出ソース

結果

問題	No.2930 Larger Mex
ユーザー	kwm_t
提出日時	2024-10-12 16:40:56
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	741 ms / 2,000 ms
コード長	4,268 bytes
コンパイル時間	3,369 ms
コンパイル使用メモリ	264,568 KB
実行使用メモリ	16,512 KB
最終ジャッジ日時	2024-10-12 16:41:26
合計ジャッジ時間	20,843 ms
ジャッジサーバーID （参考情報）	judge4 / judge

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,248 KB
testcase_02	AC	2 ms 5,248 KB
testcase_03	AC	4 ms 5,248 KB
testcase_04	AC	5 ms 5,248 KB
testcase_05	AC	5 ms 5,248 KB
testcase_06	AC	4 ms 5,248 KB
testcase_07	AC	5 ms 5,248 KB
testcase_08	AC	346 ms 5,248 KB
testcase_09	AC	292 ms 5,248 KB
testcase_10	AC	324 ms 5,248 KB
testcase_11	AC	240 ms 5,248 KB
testcase_12	AC	453 ms 8,192 KB
testcase_13	AC	365 ms 6,144 KB
testcase_14	AC	512 ms 9,728 KB
testcase_15	AC	250 ms 5,248 KB
testcase_16	AC	265 ms 5,248 KB
testcase_17	AC	333 ms 5,248 KB
testcase_18	AC	255 ms 5,248 KB
testcase_19	AC	382 ms 6,784 KB
testcase_20	AC	376 ms 6,016 KB
testcase_21	AC	340 ms 5,504 KB
testcase_22	AC	531 ms 11,264 KB
testcase_23	AC	346 ms 6,272 KB
testcase_24	AC	298 ms 5,248 KB
testcase_25	AC	638 ms 14,976 KB
testcase_26	AC	261 ms 5,248 KB
testcase_27	AC	193 ms 5,248 KB
testcase_28	AC	209 ms 5,248 KB
testcase_29	AC	218 ms 5,248 KB
testcase_30	AC	203 ms 5,248 KB
testcase_31	AC	193 ms 5,248 KB
testcase_32	AC	216 ms 5,248 KB
testcase_33	AC	225 ms 5,248 KB
testcase_34	AC	459 ms 9,088 KB
testcase_35	AC	308 ms 6,784 KB
testcase_36	AC	508 ms 10,112 KB
testcase_37	AC	408 ms 7,936 KB
testcase_38	AC	358 ms 8,192 KB
testcase_39	AC	332 ms 10,112 KB
testcase_40	AC	323 ms 9,088 KB
testcase_41	AC	259 ms 6,656 KB
testcase_42	AC	250 ms 5,248 KB
testcase_43	AC	216 ms 5,248 KB
testcase_44	AC	224 ms 5,248 KB
testcase_45	AC	216 ms 5,248 KB
testcase_46	AC	330 ms 10,496 KB
testcase_47	AC	316 ms 10,368 KB
testcase_48	AC	216 ms 8,960 KB
testcase_49	AC	321 ms 10,496 KB
testcase_50	AC	281 ms 5,248 KB
testcase_51	AC	723 ms 16,512 KB
testcase_52	AC	741 ms 16,384 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
//#include <atcoder/all>
using namespace std;
//using namespace atcoder;
//using mint = modint1000000007;
//const int mod = 1000000007;
//using mint = modint998244353;
//const int mod = 998244353;
const int INF = 1e9;
//const long long LINF = 1e18;
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep2(i,l,r)for(int i=(l);i<(r);++i)
#define rrep(i, n) for (int i = (n) - 1; i >= 0; --i)
#define rrep2(i,l,r)for(int i=(r) - 1;i>=(l);--i)
#define all(x) (x).begin(),(x).end()
#define allR(x) (x).rbegin(),(x).rend()
#define P pair<int,int>
template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; }
template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; }
#include <set>
#include <iostream> 

template<typename T>
struct RangeSet {
	// ReadMe:閉区間なことに注意
	// member
	std::set<std::pair<T, T>>st;
	T TINF;
	// constructor
	RangeSet(T lim) {
		TINF = lim + 1;
		st.emplace(-TINF, -TINF);
		st.emplace(TINF, TINF);
	}
	// cover check
	bool IsCovered(T l, T r) {
		auto ite = prev(st.lower_bound({ l + 1,l + 1 }));
		return ((ite->first <= l) && (r <= ite->second));
	}
	bool IsCovered(T x) { return IsCovered(x, x); }
	// get by covered
	std::pair<T, T> ByCovered(T l, T r) {
		auto ite = prev(st.lower_bound({ l + 1,l + 1 }));
		if ((ite->first <= l) && (r <= ite->second)) return *ite;
		return { -TINF, -TINF };
	}
	std::pair<T, T> ByCovered(T  x) { return ByCovered(x, x); }
	// getFront
	std::pair<T, T>getFront() {
		auto itr = st.begin();
		itr++;
		return *itr;
	}
	// getBack
	std::pair<T, T>getBack() {
		auto itr = st.end();
		itr--;
		itr--;
		return *itr;
	}
	// insert
	T Insert(T l, T r) {
		//insertでmargeしないver
		//st.emplace(l, r);
		//return 0;
		T sumErase = T(0);
		auto ite = prev(st.lower_bound({ l + 1,l + 1 }));
		// no need merge
		if (IsCovered(l, r))return sumErase;
		// first merge pos
		if ((ite->first <= l) && (l <= (ite->second + 1))) {
			l = ite->first;
			sumErase += ite->second - ite->first + 1;
			ite = st.erase(ite);
		}
		else ite = next(ite);
		// merge
		while (r > ite->second) {
			sumErase += ite->second - ite->first + 1;
			ite = st.erase(ite);
		}
		if (((ite->first - 1) <= r) && (r <= ite->second)) {
			r = ite->second;
			sumErase += ite->second - ite->first + 1;
			st.erase(ite);
		}
		st.emplace(l, r);
		return (r - l + 1) - sumErase;
	}
	T Insert(T x) { return Insert(x, x); }
	// erase
	T Erase(T l, T r) {
		T ret = T(0);
		auto get = ByCovered(l, r);
		if ((-TINF) != get.first) {
			st.erase(get);
			if (get.first != l)st.emplace(get.first, l - 1);
			if (get.second != r)st.emplace(r + 1, get.second);
			ret += r - l + 1;
			return ret;
		}
		auto ite = prev(st.lower_bound({ l + 1,l + 1 }));
		// first erase
		if ((ite->first <= l) && (l <= ite->second)) {
			ret += ite->second - l + 1;
			if (ite->first != l)st.emplace(ite->first, l - 1);
			ite = st.erase(ite);
		}
		else ite = next(ite);
		// erase
		while (ite->second <= r) {
			ret += ite->second - ite->first + 1;
			ite = st.erase(ite);
		}
		// last
		if ((ite->first <= r) && (r <= ite->second)) {
			ret += r - ite->first + 1;
			if (ite->second != r) st.emplace(r + 1, ite->second);
			st.erase(ite);
		}
		return ret;
	}
	T Erase(T x) { return Erase(x, x); }
	// range count
	int size() { return st.size() - 2; }
	// mex
	T Mex(T x = 0) {
		if (!IsCovered(x)) return x;
		auto ite = prev(st.lower_bound({ x + 1, x + 1 }));
		return ite->second + 1;
	}
	// debug
	void Debug() {
		for (auto[l, r] : st)std::cout << l << " " << r << std::endl;
	}
};
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n, m; cin >> n >> m;
	vector<int>a(n);
	rep(i, n)cin >> a[i];
	RangeSet<int>rst(INF);
	map<int, int>mp;
	int j = 0;
	vector<int>ans(n + 2);
	rep(i, n) {
		while (j < n && rst.Mex() < m) {
			if (0 == mp[a[j]])rst.Insert(a[j]);
			mp[a[j]]++;
			j++;
		}
		if (rst.Mex() >= m) {
			//cout << j - i << " " << n - i + 1 << endl;
			ans[j - i]++;
			ans[n - i + 1]--;
		}
		mp[a[i]]--;
		if (0 == mp[a[i]])rst.Erase(a[i]);
		j = max(j, i + 1);
	}
	rep(i, n + 1)ans[i + 1] += ans[i];
	rep2(i, 1, n + 1)cout << ans[i] << endl;
	return 0;
}

yukicoder

結果

テストケース

ソースコード