USTCOJ 1215 未知星球

题目链接：http://acm.ustc.edu.cn/ustcoj/problem.php?id=1215

个人觉得是一道好题，大意是给一串数字，循环的（首尾实际上是相接的），求最长的一段使得这段里最大的值减最小值不超过k。下面给出我想的一个nlogn的算法，主要是用ST求RMQ问题，加上二分猜答案思想，我们的序列长度最大值已知，且一旦一个较小的k不可能是解，则更大的更不可能（因为判断一个k是不是解是从头到尾遍历的。）这样二分是logn，验证是这样的，我们利用ST算法预处理，在o（n）时间里求一段区间max - min， 加上最后遍历是n所以是o（n），所以合在一起是nlogn。注意遍历判读是否是解需要考略循环，首尾相接。

代码如下：

#include<iostream>#include<cstring>#define mem(a) memset(a, 0, sizeof(a))using namespace std;const int N = 20005;int a[N], dm[N][20], dx[N][20]; //m min, x maxvoid pre(int n){int i, j;for (i = 0; i < n; ++i) dm[i][0] = dx[i][0] = a[i];for (j = 1; (1 << j) <= n; ++j)for (i = 0; i + (1 << j) - 1 < n; ++i)dm[i][j] = min(dm[i][j - 1], dm[i + (1 << (j - 1))][j - 1]), dx[i][j] = max(dx[i][j - 1], dx[i + (1 << (j - 1))][j - 1]);}int qx(int s, int e){int k = 0;while ((1 << (k + 1)) <= e - s + 1) k++;return max(dx[s][k], dx[e - (1 << k) + 1][k]);}int qm(int s, int e){int k = 0;while ((1 << (k + 1)) <= e - s + 1) k++;return min(dm[s][k], dm[e - (1 << k) + 1][k]);}int ask(int len, int n, int k){int i, t;for (i = 0; i < n; ++i){if (i + len - 1 < n) t = qx(i, i + len - 1) - qm(i, i + len - 1);else t = max(qx(i, n - 1), qx(0, len + i - n - 1)) - min(qm(i, n - 1), qm(0, len + i - n - 1));if (t <= k) return 1;}return 0;}int bs(int s, int e, int n, int k){int m = (e + s) >> 1;if (e <= s + 1) return s;else{if (ask(m, n, k)) return bs(m, e, n, k);else return bs(s, m, n, k);}}int main(){int n, k, i;while (cin >> n >> k, n | k){mem(dx), mem(dm);for (i = 0; i < n; ++i) cin >> a[i];pre(n);cout << bs(1, n + 1, n, k) << endl;}return 0;}