#869823 (C++17) No.919 You Are A Project Manager

提出ソース

結果

問題	No.919 You Are A Project Manager
ユーザー	kwm_t
提出日時	2023-05-16 23:28:13
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	1,215 ms / 3,000 ms
コード長	4,295 bytes
コンパイル時間	2,878 ms
コンパイル使用メモリ	228,108 KB
実行使用メモリ	17,684 KB
最終ジャッジ日時	2024-12-14 20:54:45
合計ジャッジ時間	28,903 ms
ジャッジサーバーID （参考情報）	judge1 / judge5

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1,196 ms 16,792 KB
testcase_01	AC	2 ms 6,820 KB
testcase_02	AC	2 ms 6,820 KB
testcase_03	AC	2 ms 6,820 KB
testcase_04	AC	125 ms 6,816 KB
testcase_05	AC	145 ms 6,816 KB
testcase_06	AC	6 ms 6,820 KB
testcase_07	AC	274 ms 8,704 KB
testcase_08	AC	41 ms 6,820 KB
testcase_09	AC	27 ms 6,816 KB
testcase_10	AC	57 ms 6,816 KB
testcase_11	AC	53 ms 6,820 KB
testcase_12	AC	2 ms 6,820 KB
testcase_13	AC	70 ms 6,820 KB
testcase_14	AC	4 ms 6,816 KB
testcase_15	AC	16 ms 6,816 KB
testcase_16	AC	118 ms 6,820 KB
testcase_17	AC	1,154 ms 17,432 KB
testcase_18	AC	1,126 ms 17,168 KB
testcase_19	AC	1,215 ms 16,660 KB
testcase_20	AC	913 ms 16,016 KB
testcase_21	AC	1,186 ms 16,788 KB
testcase_22	AC	304 ms 9,212 KB
testcase_23	AC	289 ms 8,832 KB
testcase_24	AC	327 ms 9,600 KB
testcase_25	AC	302 ms 9,084 KB
testcase_26	AC	855 ms 15,248 KB
testcase_27	AC	727 ms 14,128 KB
testcase_28	AC	980 ms 16,024 KB
testcase_29	AC	1,132 ms 17,556 KB
testcase_30	AC	1,135 ms 17,300 KB
testcase_31	AC	1,119 ms 17,684 KB
testcase_32	AC	1,127 ms 17,296 KB
testcase_33	AC	335 ms 9,472 KB
testcase_34	AC	339 ms 9,472 KB
testcase_35	AC	1,007 ms 16,656 KB
testcase_36	AC	1,009 ms 16,664 KB
testcase_37	AC	1,006 ms 16,660 KB
testcase_38	AC	1,039 ms 16,664 KB
testcase_39	AC	2 ms 6,820 KB
testcase_40	AC	17 ms 6,816 KB
testcase_41	AC	4 ms 6,816 KB
testcase_42	AC	6 ms 6,820 KB
testcase_43	AC	9 ms 6,820 KB
testcase_44	AC	25 ms 6,816 KB
testcase_45	AC	6 ms 6,820 KB
testcase_46	AC	20 ms 6,820 KB
testcase_47	AC	301 ms 8,832 KB
testcase_48	AC	298 ms 8,960 KB
testcase_49	AC	323 ms 9,212 KB
testcase_50	AC	307 ms 9,212 KB
testcase_51	AC	271 ms 8,836 KB
testcase_52	AC	191 ms 7,420 KB
testcase_53	AC	268 ms 8,496 KB
testcase_54	AC	17 ms 6,816 KB
testcase_55	AC	2 ms 6,816 KB
testcase_56	AC	2 ms 6,820 KB
testcase_57	AC	2 ms 6,820 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 endl "\n"
#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; }
template<typename T>
struct Median{
	multiset<T> x, y;
	Median() {}
	void add(T val) {
		x.insert(val);
		y.insert(*x.rbegin());
		x.erase(x.find(*x.rbegin()));
		x.insert(*y.begin());
		y.erase(y.begin());
		if (x.size() > y.size() + 1) {
			y.insert(*x.rbegin());
			x.erase(x.find(*x.rbegin()));
		}
	}
	// 存在は保証されている
	void del(T val) {
		if (x.count(val)) {
			x.erase(x.find(val));
			if (x.size() < y.size()) {
				x.insert(*y.begin());
				y.erase(y.begin());
			}
		}
		else  {
			y.erase(y.find(val));
			if (x.size() > y.size() + 1) {
				y.insert(*x.rbegin());
				x.erase(x.find(*x.rbegin()));
			}
		}
	}
	T get() {
		if (x.empty())return -1;//error
		return *x.rbegin();
	}
};
struct Mo {
	int width;
	vector<int>left, right, order;
	vector<int>ans;
	Median<int> median;
	//vector<long long>horder;
	int n, q;
	Mo(int _n, int _q) :n(_n), q(_q) {
		width = (int)sqrt(_n);
		order.resize(q);
		iota(all(order), 0);
		ans.resize(q);
	}
	// add,query[l,r)
	void query(vector<int> &l, vector<int> &r) {
		swap(left, l);
		swap(right, r);
		//horder.resize(q);
		//rep(i, q) horder[i] = hilbertorder(left[i], right[i]);
	}
	// run
	void run(const vector<int>&data) {
		sort(all(order), [&](int lh, int rh) {
			//return horder[lh] < horder[rh];
			int lblock = left[lh] / width;
			int rblock = left[rh] / width;
			if (lblock != rblock)return lblock < rblock;
			if (lblock & 1) return right[lh] < right[rh];
			return right[lh] > right[rh];
			});
		int nl = 0, nr = 0;
		auto add = [&](int idx) {
			median.add(data[idx]);
		};
		auto del = [&](int idx) {
			median.del(data[idx]);
		};
		auto ansset = [&](int idx) {
			ans[idx] = median.get();
		};
		for (auto idx : order) {
			while (nl > left[idx])add(--nl);
			while (nr < right[idx])add(nr++);
			while (nl < left[idx])del(nl++);
			while (nr > right[idx])del(--nr);
			ansset(idx);
		}
	}
	// TLEするときはこっちを使うと通るかも
	/*const int logn = 20;
	const int maxn = 1 << logn;
	long long hilbertorder(int x, int y) {
		long long d = 0;
		for (int s = 1 << logn - 1; s; s >>= 1) {
			bool rx = x & s, ry = y & s;
			d = d << 2 | rx * 3 ^ static_cast<int>(ry);
			if (ry) continue;
			if (rx) {
				x = maxn - x;
				y = maxn - y;
			}
			swap(x, y);
		}
		return d;
	}*/
};
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n; cin >> n;
	vector<int>a(n);
	rep(i, n)cin >> a[i];
	long long ans = 0;
	map<P, int>mp;
	vector<int>l, r;
	rep2(i, 1, n + 1) {
		for (int j = 0; j + i <= n; j += i) {
			l.push_back(j);
			r.push_back(j + i);
			mp[{j, j + i}] = mp.size();
		}
		if (0 == n % i)continue;
		for (int j = n; j - i >= 0; j -= i) {
			l.push_back(j - i);
			r.push_back(j);
			mp[{j - i, j}] = mp.size();
		}
	}
	Mo mo(n, mp.size());
	mo.query(l, r);
	mo.run(a);
	rep2(i, 1, n + 1) {
		vector<int>l, r;
		for (int j = 0; j + i <= n; j += i) {
			int idx = mp[{j, j + i}];
			l.push_back(mo.ans[idx]);
		}
		for (int j = n; j - i >= 0; j -= i) {
			int idx = mp[{j - i, j}];
			r.push_back(mo.ans[idx]);
		}
		vector<long long>sl(l.size() + 1), sr(r.size() + 1);
		rep(j, l.size())sl[j + 1] = sl[j] + l[j];
		rep(j, l.size())chmax(sl[j + 1], sl[j]);
		rep(j, r.size())sr[j + 1] = sr[j] + r[j];
		rep(j, r.size())chmax(sr[j + 1], sr[j]);
		int mx = n / i;
		rep(j, mx + 1) {
			long long sub = 0;
			sub += sl[j] - sl[0];
			int k = mx - j;
			sub += sr[k] - sr[0];
			chmax(ans, sub * i);
		}
	}
	cout << ans << endl;
	return 0;
}

yukicoder

結果

テストケース

ソースコード