poj2182 Lost Cows [osily]

Lost Cows

这道题题意很明白，开始用n*nlogn的方法毫无悬念地TLE了，又想不出更好的方法，虽然Discuss上有人用类似的方法确实可以，但更多人用线段树做的，于是去看了看。

#include <stdio.h> #include <string.h> struct node { int l; int r; int len; } a[2 * 10005]; //数组模拟线段树 void creat_tree(int l, int r, int p) { a[p].l = l; a[p].r = r; a[p].len = r - l + 1; if (l == r) return; else { int mid = (l + r) / 2; creat_tree(l, mid, 2 * p); creat_tree(mid + 1, r, 2 * p + 1); } } int order(int p, int n) { a[p].len--; if (a[p].l == a[p].r) return a[p].l; else if (a[2 * p].len >= n) return order(2 * p, n); else return order(2 * p + 1, n - a[2 * p].len); } int main() { int n, i, b[8005], re[8005]; while (scanf("%d", &n) == 1) { for (i = 1; i < n; i++) scanf("%d", &b[i]); creat_tree(1, n, 1); for (i = n - 1; i > 0; i--) re[i] = order(1, b[i] + 1); re[0] = order(1, 1); for (i = 0; i < n; i++) printf("%d/n", re[i]); } return 0; }

这是一个标准的线段树，而此题的树结点多了一个长度的元素，主要是为了查找的。

由于数据规模不是太大，此树类似于完全二叉树，故可以用数组模拟。

开始时要建树，从上到下递归实现，这个只操作一回便可。

接下来根据题意，将输入数据从后到前依次分析，每次都是找到按大小排第k的，并将其删除。重点就在这里，开始也是这里不理解。

每次都是从根结点开始，根据数据不停判断属于左树还是右树并递归向下，直到叶子结点，所在的数就是结果。所谓删除是用改变len实现的，因为一旦len减少，找时就要右移，将找过的叶子结点跳过。

看了些资料，发现线段树很有用，有时间要专门研究下。