「 题解 」P2487 [SDOI2011]拦截导弹

简单题意

给定 \(n\) 个数对 \((h_i, v_i)\)

求:

  1. 最长不上升子序列的长度。
  2. 对于每个 \(i\),分别求出包含数对 \((h_i, v_i)\) 的最长上升子序列的个数和最长不上升子序列的个数和的比。

数据范围\(1 \leq n \leq 5 \times {10} ^ 4\)\(\forall 1 \leq i \leq n, 1 \leq h_i, v_i \leq {10}^9\)

分析

问题 \(1\)

先考虑 \(O(n^2)\) 做法,本质与一维的相同。

定义 \(f_i\) 表示以数对 \((h_i, v_i)\) 结尾的最长不上升子序列的长度。

那么有

\[\begin{aligned} f_i & = \max(1, \max\{ f_j + 1 \}), h_i \leq h_j \wedge v_i \leq v_j \\ \end{aligned} \]

推出了式子后考虑优化,可以使用 \(\texttt{cdq}\) 分治。

对于一个区间 \(\texttt{[l, r]}\) 的转移。

\(\texttt{int mid = (l + r) >> 1}\)

先递归 \(\texttt{cdq(l, mid)}\)

假设 \(\texttt{[l, mid]}\) 已经求出正确答案,即 \(f_{\texttt{l} \sim \texttt{r}}\) 都是正确的。

考虑如何转移,即 \(\texttt{[l, mid]}\)\(\texttt{[mid + 1, r]}\) 的贡献。

与三维偏序一样,合并 \(h\),以 \(v\) 为下标把 \(f\) 存放在树状数组中。

只不过这里的 \(f\) 需要取最大值。

最后递归 \(\texttt{cdq(mid + 1, r)}\)

因为是最后递归 \(\texttt{cdq(mid + 1, r)}\),所以询问不能真正合并,在结束时需要还原成原来的顺序。

问题 \(2\)

问题 \(1\) 解决后问题 \(2\) 就简单了。

只需要在树状数组中再维护一个统计个数数组即可。

  • 修改的值大于当前最大值就修改。
  • 修改的值等于当前最大值就累加。

因为求的是 包含 数对 \((h_i, v_i)\) 的最长不上升子序列的个数,所以还需要反着求一遍最长不上升子序列的个数。

两个数相乘再除以总数就是答案,前提是这个数对被包含在至少一个最长不上升子序列中。

温馨提示,个数可能会超过 \(\texttt{long long}\) 的范围,建议使用 \(\texttt{double}\)

\(\texttt{code}\)

#include 
#include 
#include 
#include 
#include 

int rint() {
	int x = 0, fx = 1;
	char c = getchar();
	
	while (c < '0' || c > '9') {
		fx ^= (c == '-');
		c = getchar();
	}
	
	while ('0' <= c && c <= '9') {
		x = (x << 3) + (x << 1) + (c ^ 48);
		c = getchar();
	}
	
	if (!fx) {
		return -x;
	}
	
	return x;
}

void read(int &x) {
	x = rint();
}

template
void read(int &x, Ts &...rest) {
	read(x);
	read(rest...);
}

int Max(int u, int v) {
	return (u > v) ? u : v;
}

int Min(int u, int v) {
	return (u < v) ? u : v;
}

const int MAX_n = 5e4;

int n, Time; // Time 是时间戳优化树状数组,可以不用清空 
int dp1[MAX_n + 5]; // 正 
int dp2[MAX_n + 5]; // 反 
int vis[MAX_n + 5]; // 树状数组时间戳 
int Bit[MAX_n + 5];
double bit[MAX_n + 5];
double num1[MAX_n + 5]; // 正 
double num2[MAX_n + 5]; // 反 

std::vector lsh; // 离散化 v 

struct Missile {
	int idx, h, v;
} q[MAX_n + 5];
Missile tmp[MAX_n + 5];

bool cmph1(Missile x, Missile y) {
	return x.h > y.h;
}

bool cmph2(Missile x, Missile y) {
	return x.h < y.h;
}

int lowbit(int x) {
	return x & (-x);
}

void add(int k, int x, double y) {
	while (k <= n) {
		if (vis[k] != Time) {
			vis[k] = Time;
			Bit[k] = x;
			bit[k] = y;
		} else {
			if (x > Bit[k]) {
				Bit[k] = x;
				bit[k] = y;
			} else if (x == Bit[k]) {
				bit[k] += y;
			}
		}
		
		k += lowbit(k);
	}
}

std::pair ask(int k) {
	int resmax = 0;
	double ressum = 0.0;
	
	while (k > 0) {
		if (vis[k] == Time) {
			if (Bit[k] > resmax) {
				resmax = Bit[k];
				ressum = bit[k];
			} else if (Bit[k] == resmax) {
				ressum += bit[k];
			}
		}
		
		k -= lowbit(k);
	}
	
	return std::make_pair(resmax, ressum);
}

void merge1(int L1, int R1, int L2, int R2) {
	++Time;
	int i = L1, j = L2;
	
	for (int k = L1; k <= R2; k++) {
		tmp[k] = q[k];
	}
	
	std::sort(q + L1, q + R1 + 1, cmph1);
	std::sort(q + L2, q + R2 + 1, cmph1);
	
	while (i <= R1 || j <= R2) {
		if (i <= R1 && (j > R2 || q[i].h >= q[j].h)) {
			add((int)lsh.size() + 1 - q[i].v, dp1[q[i].idx], num1[q[i].idx]);
			i++;
		} else {
			std::pair now = ask((int)lsh.size() + 1 - q[j].v);
			
			if (now.first + 1 > dp1[q[j].idx]) {
				dp1[q[j].idx] = now.first + 1;
				num1[q[j].idx] = now.second;
			} else if (now.first + 1 == dp1[q[j].idx]) {
				num1[q[j].idx] += now.second;
			}
			
			j++;
		}
	}
	
	for (int k = L1; k <= R2; k++) {
		q[k] = tmp[k];
	}
}

void cdq1(int L, int R) {
	if (L == R) {
		return ;
	}
	
	int Mid = (L + R) >> 1;
	cdq1(L, Mid);
	merge1(L, Mid, Mid + 1, R);
	cdq1(Mid + 1, R);
}

void merge2(int L1, int R1, int L2, int R2) {
	++Time;
	int i = L1, j = L2;
	
	for (int k = L1; k <= R2; k++) {
		tmp[k] = q[k];
	}
	
	std::sort(q + L1, q + R1 + 1, cmph2);
	std::sort(q + L2, q + R2 + 1, cmph2);
	
	while (i <= R1 || j <= R2) {
		if (i <= R1 && (j > R2 || q[i].h <= q[j].h)) {
			add(q[i].v, dp2[q[i].idx], num2[q[i].idx]);
			i++;
		} else {
			std::pair now = ask(q[j].v);
			
			if (now.first + 1 > dp2[q[j].idx]) {
				dp2[q[j].idx] = now.first + 1;
				num2[q[j].idx] = now.second;
			} else if (now.first + 1 == dp2[q[j].idx]) {
				num2[q[j].idx] += now.second;
			}
			
			j++;
		}
	}
	
	for (int k = L1; k <= R2; k++) {
		q[k] = tmp[k];
	}
}

void cdq2(int L, int R) {
	if (L == R) {
		return ;
	}
	
	int Mid = (L + R) >> 1;
	cdq2(L, Mid);
	merge2(L, Mid, Mid + 1, R);
	cdq2(Mid + 1, R);
}

signed main() {
	n = rint();
	
	for (int i = 1; i <= n; i++) {
		read(q[i].h, q[i].v);
		q[i].idx = i;
		lsh.push_back(q[i].v);
	}
	
	std::sort(lsh.begin(), lsh.end());
	lsh.resize(std::unique(lsh.begin(), lsh.end()) - lsh.begin());
	
	for (int i = 1; i <= n; i++) {
		q[i].v = std::lower_bound(lsh.begin(), lsh.end(), q[i].v) - lsh.begin() + 1;
		dp1[i] = dp2[i] = 1;
		num1[i] = num2[i] = 1.0;
	}
	
	cdq1(1, n);
	
	for (int i = n / 2; i >= 1; i--) {
		std::swap(q[i], q[n + 1 - i]);
	}
	
	cdq2(1, n);
	int res = 0;
	double sum = 0;
	
	for (int i = 1; i <= n; i++) {
		if (dp1[i] > res) {
			res = dp1[i];
			sum = num1[i];
		} else if (dp1[i] == res) {
			sum += num1[i];
		}
	}
	
	printf("%d\n", res);
	
	for (int i = 1; i <= n; i++) {
		if (dp1[i] + dp2[i] - 1 != res) {
			printf("0.00000");
		} else {
			printf("%.5f", 1.0 * num1[i] * num2[i] / sum);
		}
		
		putchar((i == n) ? '\n' : ' ');
	}
	
	return 0;
}

你可能感兴趣的:(「 题解 」P2487 [SDOI2011]拦截导弹)