这两题差不多一个意思, 2828加了一个ID;
poj 2182
题意简述:
有N头牛,编号为1~N,乱序排成一列,现在已知每头牛前面有多少头牛比它的编号小,求排队后从前往后数,每头牛的编号。
暴力思路:从后往前求,定义一个a[I]数组表示前面有i头牛比它编号小,此时的位置,每求出一头牛的具体位置,就要更新a数组;
代码:
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #include <cmath> #define MAXN 8008 using namespace std; int a[MAXN], pre[MAXN], res[MAXN]; int main() { int n; scanf("%d", &n); int m = n-1; for(int i = 0; i < m; i++) { a[i] = i+1; scanf("%d", &pre[i]); } a[m] = n; for(int i = n-2; i >= 0; i--) { res[i+1] = a[pre[i]]; for(int j = pre[i]; j < m; j++) a[j] = a[j+1]; m--; } res[0] = a[0]; for(int i = 0; i < n; i++) printf("%d\n", res[i]); return 0; }线段树做法:
线段树里面的v表示这个区间内未被删除的数字个数;
具体看代码:
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #include <cmath> #define MAXN 8008 using namespace std; struct Node{ int l, r, v; } tree[3*MAXN];//是不是有人疑问这里,你举个例子就好了; int res[MAXN], a[MAXN]; void build(int k, int l, int r) { tree[k].l = l; tree[k].r = r; tree[k].v = r-l+1; if (l == r) return; int mid = (l+r)/2; build(k*2, l, mid); build(k*2+1, mid+1, r); return ; } int query(int j, int k) { tree[j].v--; // 节点长度减一 if (tree[j].l == tree[j].r) return tree[j].l; if (tree[j*2].v >= k) //如果左子树的长度大于k就搞左子树 return query(j*2, k); else return query(j*2+1, k-tree[2*j].v); //搞右子树的话就要减去左子树的长度; } int main() { int n; scanf("%d", &n); for(int i = 2; i <= n; i++) scanf("%d", a+i); a[1] = 0; build(1, 1, n); for(int i = n; i > 0; i--) res[i] = query(1, a[i]+1); for(int i = 1; i <= n; i++) printf("%d\n", res[i]); return 0; } poj 2828,不多bb,一样的; 代码: #include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #include <cmath> #define MAXN 200008 using namespace std; struct Node{ int l, r, v; } tree[3*MAXN]; int a[MAXN], val[MAXN], res[MAXN]; void build(int k, int l, int r) { tree[k].l = l; tree[k].r = r; tree[k].v = r-l+1; if (l == r) return ; int mid = (l+r)/2; build(k*2, l, mid); build(k*2+1, mid+1, r); return ; } int query(int j, int k) { tree[j].v--; if (tree[j].l == tree[j].r) return tree[j].l; if (tree[j*2].v >= k) return query(j*2, k); else return query(j*2+1, k-tree[2*j].v); } int main() { int n; while (~scanf("%d", &n)) { for(int i = 1; i <= n; i++) scanf("%d%d", a+i, val+i); build(1, 1, n); for(int i = n; i > 0; i--) { int tmp = query(1, a[i]+1); res[tmp] = val[i]; //就这里不同,要想一下下 } for(int i = 1; i <= n; i++) printf("%d ", res[i]); puts(" "); } return 0; }
