結果

問題 No.919 You Are A Project Manager
コンテスト
ユーザー QCFium
提出日時 2019-11-01 17:28:31
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,937 bytes
コンパイル時間 1,894 ms
コンパイル使用メモリ 179,324 KB
実行使用メモリ 19,520 KB
最終ジャッジ日時 2024-09-14 22:46:38
合計ジャッジ時間 11,058 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other TLE * 1 -- * 54
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

int ri() {
	int n;
	scanf("%d", &n);
	return n;
}

// dummy one
struct AVL {
	std::vector<int> data;
	void insert(int n) {
		data.push_back(n);
		std::sort(data.begin(), data.end());
	}
	void remove(int n) {
		auto itr = std::find(data.begin(), data.end(), n);
		assert(itr != data.end());
		data.erase(itr);
	}
	int operator [] (int i) {
		return data[i];
	}
	size_t size() { return data.size(); }
};

int main() {
	int n = ri();
	int a[n];
	for (int i = 0; i < n; i++) a[i] = ri();
	std::vector<int64_t> median_left[n + 1];
	std::vector<int64_t> median_right[n + 1];
	for (int i = 0; i <= n; i++) {
		median_left[i].resize(n / (i + 1));
		median_right[i].resize(n / (i + 1));
	}
	int sq = sqrt(n);
	for (int i = 0; i < sq; i++) {
		{
			int head = 0;
			int tail = 0;
			AVL set;
			for (int j = 0; i < (int) median_left[j].size(); j++) {
				// j + 1
				while (head < (j + 1) * (i + 1)) set.insert(a[head++]);
				while (tail < (j + 1) * i) set.remove(a[tail++]);
				assert(set.size());
				median_left[j][i] = set[(set.size() - 1) / 2];
			}
		}
		{
			int head = n - 1;
			int tail = n - 1;
			AVL set;
			for (int j = 0; i < (int) median_left[j].size(); j++) {
				// j + 1
				while (head >= n - (j + 1) * (i + 1)) set.insert(a[head--]);
				while (tail >= n - (j + 1) * i) set.remove(a[tail--]);
				assert(set.size());
				median_right[j][i] = set[(set.size() - 1) / 2];
			}
		}
	}
	for (int i = 0; i <= n; i++) {
		for (int j = sq; j < (int) median_left[i].size(); j++) {
			std::vector<int> sorted;
			for (int k = j * (i + 1); k < (j + 1) * (i + 1); k++) sorted.push_back(a[k]);
			std::sort(sorted.begin(), sorted.end());
			median_left[i][j] = sorted[(sorted.size() - 1) / 2];
		}
		for (int j = sq; j < (int) median_right[i].size(); j++) {
			std::vector<int> sorted;
			for (int k = n - (j + 1) * (i + 1); k < n - j * (i + 1); k++) sorted.push_back(a[k]);
			std::sort(sorted.begin(), sorted.end());
			median_right[i][j] = sorted[(sorted.size() - 1) / 2];
		}
	}
	int64_t res = 0;
	for (int i = 0; i < n; i++) {
		int max_index_sum = (int) median_left[i].size() - 2;
		for (int j = 0; j + 1 < (int) median_left[i].size(); j++) median_left[i][j + 1] += median_left[i][j];
		for (int j = 0; j + 1 < (int) median_left[i].size(); j++) median_left[i][j + 1] = std::max(median_left[i][j + 1], median_left[i][j]);
		for (int j = 0; j + 1 < (int) median_right[i].size(); j++) median_right[i][j + 1] += median_right[i][j];
		for (int j = 0; j + 1 < (int) median_right[i].size(); j++) median_right[i][j + 1] = std::max(median_right[i][j + 1], median_right[i][j]);
		int64_t cur = 0;
		for (int j = 0; j + 1 < (int) median_left[i].size(); j++) cur = std::max(cur, median_left[i][j] + median_right[i][max_index_sum - j]);
		cur = std::max(cur, median_left[i].back());
		cur = std::max(cur, median_right[i].back());
		res = std::max(res, cur * (i + 1));
	}
	std::cout << res << std::endl;
	
	return 0;
}
0