校内模拟考 (一)

第一题

一个裸裸的树形DP。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,t,tot,s,a[N],h[N];
int cnt,fro[N],to[N],nxt[N];
inline int read() {
    int x=0; char c=getchar();
    while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-'0',c=getchar();
    return x;
}
void add(int x,int y) {
    to[++cnt]=y,nxt[cnt]=fro[x]; fro[x]=cnt;
}

void DP(int u) {
    if(fro[u]==0) {a[u]=1; return ;}
    for(int i=fro[u];i;i=nxt[i]) DP(to[i]);
    tot=0; memset(h,0,sizeof(h));
    for(int i=fro[u];i;i=nxt[i]) h[++tot]=a[to[i]];
    sort(h+1,h+tot+1);
    for(int j=1;j<=ceil((double)tot*t/100);j++) a[u]+=h[j];
}

int main() {
//  freopen("pa.in","r",stdin);
//  freopen("pa.out","w",stdout);
    n=read(),t=read();
    for(int i=1;i<=n;i++) {int x=read(); add(x,i);}
    DP(0); 
    printf("%d",a[0]);
}

第二题

理一理思路

1）将人按身高从矮到高排序，按这个顺序处理时无论当前怎么放，都不会对后面更高的产生影响。

2）线段树：统计当前答案序列$1$~$x$的位置有几个未确定的 (即有几个更高的)。

3）字典序最小：在处理时，尽量把当前数往左放。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,ans[N];
struct node {int h,x;}a[N];
struct tree {int l,r,s;}f[N*3];
inline int read() {
    int x=0; char c=getchar();
    while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-'0',c=getchar();
    return x;
}
bool cmp(node x1,node x2) {return x1.h<x2.h;}

void build(int p,int l,int r) {
    f[p].l=l,f[p].r=r;
    if(l==r) {f[p].s=1; return ;}
    int Mid=(l+r)>>1;
    build(p*2,l,Mid);
    build(p*2+1,Mid+1,r);
    f[p].s=f[p*2].s+f[p*2+1].s;
}

void find(int p,int num,int h) {
    f[p].s--;
    if(f[p].l==f[p].r) {ans[f[p].l]=h; return ;}
    if(num<f[p*2].s) find(p*2,num,h);
    else find(p*2+1,num-f[p*2].s,h);
}

int main() {
//  freopen("pb.in","r",stdin);
//  freopen("pb.out","w",stdout);
    n=read();
    for(int i=1;i<=n;i++)
     a[i].h=read(),a[i].x=read();
    sort(a+1,a+n+1,cmp);
    for(int i=1;i<=n;i++) 
     if(a[i].x>n-i) return printf("impossible"),0;
    build(1,1,n);
    for(int i=1;i<=n;i++)
     find(1,min(a[i].x,n-i-a[i].x),a[i].h);
    for(int i=1;i<=n;i++) printf("%d ",ans[i]);
}