1. 程式人生 > 其它 >題解 CF1539F Strange Array

題解 CF1539F Strange Array

賽場上順利地做完前面四題就來看 F 了,但是一直沒有什麼思路,現在發現還是很可做的。

題目裡要求的“和中心的距離”這個條件不太好求,因為要中心就得知道區間長度,而資料範圍有 \(2 \times 10^5\), 不支援列舉長度後再幹什麼了,所以肯定要經過適當的轉化。

如果該數在區間中心的右邊,那麼它離中心的距離應該是區間內小於等於它的個數減去大於等於它的個數再除以二,這點用 \(cntl, cntr, cnte\) 表示比其小的/大的/相等的就可以得到。
數在左邊同樣可以得到一個式子。

那麼這個東西怎麼處理呢?一段區間以內求比其小的大的似乎是需要樹套樹的,而且要求這樣一個和的最大值也不好做。
先解決一下比一個數小的問題。權值線段樹就算了,可以直接將數排個序,然後處理每個之前將把比它大的標記掉就行了。所以只剩下區間的問題。
這個區間上一些個數的差可以按套路把後面減的標記為 \(-1\)

, 然後就只需要求跨過這個點的所有區間中和最大的。可以用線段樹來解決。具體地,若我們要跨過的點在 \(x\), 那麼就是要求 \(1\to x-1\) 中最大的字尾和 \(x+1 \to n\) 中最大的字首的和。用線段樹維護區間最大字首和字尾就可以了,寫成結構體可能會方便一些。

一些細節看程式碼吧。

#include <cstdio>
#include <vector>
const int N = 200005;
int n, ans[N], x;
struct twt {
	int sum, pre, suf;
	twt(int x = 0, int y = 0, int z = 0) { sum = x, pre = y, suf = z; }
	friend twt operator + (twt a, twt b) {
		return twt(a.sum + b.sum, std::max(a.pre, b.pre + a.sum), std::max(b.suf, a.suf + b.sum));
	}
} t[N<<2];
std::vector<int> pos[N];
void Update(int p, int l, int r, int x, int y) {
	if(l == r) {
		t[p].pre = std::max(0, y), t[p].suf = std::max(0, y);
		t[p].sum = y;
		return;
	}
	int mid = l + (r-l) / 2;
	if(x <= mid) Update(p+p, l, mid, x, y);
	else Update(p+p+1, mid+1, r, x, y);
	t[p] = t[p+p] + t[p+p+1];
}
twt Query(int p, int l, int r, int x, int y) {
	if(x > y) return twt(0, 0, 0);
	if(l == x && r == y) return t[p];
	int mid = l + (r-l) / 2;
	if(y <= mid) return Query(p+p, l, mid, x, y);
	else if(x > mid) return Query(p+p+1, mid+1, r, x, y);
	else return Query(p+p, l, mid, x, mid) + Query(p+p+1, mid+1, r, mid+1, y);
}
void solve1() {
	for(int i = 1; i <= n; i++) Update(1, 1, n, i, 1);
	for(int i = 1; i <= n; i++) {
		for(int j = 0; j < (signed)pos[i].size(); j++) {
			int v = pos[i][j];
			int an = Query(1, 1, n, 1, v-1).suf + 1 + Query(1, 1, n, v+1, n).pre;
			ans[v] = std::max(ans[v], an/2);
		}
		for(int j = 0; j < (signed)pos[i].size(); j++) Update(1, 1, n, pos[i][j], -1);
	}
}
void solve2() {
	for(int i = 1; i <= n; i++) Update(1, 1, n, i, 1);
	for(int i = n; i >= 1; i--) {
		for(int j = 0; j < (signed)pos[i].size(); j++) {
			int v = pos[i][j];
			int an = Query(1, 1, n, 1, v-1).suf + 1 + Query(1, 1, n, v+1, n).pre;
			ans[v] = std::max(ans[v], (an-1)/2);
		}
		for(int j = 0; j < (signed)pos[i].size(); j++) Update(1, 1, n, pos[i][j], -1);
	}
}
int main() {
	scanf("%d", &n);
	for(int i = 1; i <= n; i++) scanf("%d", &x), pos[x].push_back(i);
	solve1(), solve2();
	for(int i = 1; i <= n; i++) printf("%d ", ans[i]);
	return 0;
}