这两题差不多一个意思, 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;
}