【题解】晋升者计数 Promotion Counting [USACO 17 JAN] [P3605]

【题解】晋升者计数 Promotion Counting [USACO 17 JAN] [P3605]


奶牛们又一次试图创建一家创业公司,还是没有从过去的经验中吸取教训。!牛是可怕的管理者!


【题目描述】

奶牛从 \(1\) ~ \(N(1≤N≤1e5)\) 进行了编号,把公司组织成一棵树,\(1\)号奶牛作为总裁(树的根节点)。除总裁以外的每头奶牛都有且仅有唯一的一个的上司(即它在树上的父结点)。每一头牛\(i\)都有一个不同的能力指数 \(p(i)\),描述了她对其工作的擅长程度。如果奶牛 \(i\) 是奶牛 \(j\) 的父亲或祖先节点,那么奶牛 \(j\) 即为 \(i\) 的下属。

不幸地是,奶牛们发现经常有一个上司比她的下属能力低的情况,在这种情况下,上司应当考虑晋升她的一些下属。你的任务是对于公司的中的每一头奶牛 \(i\),计算其满足 \(p(j)>p(i)\) 的下属 \(j\) 的数量。

【输入】

输入的第一行包括一个整数 \(N\)。
接下来的 \(N\) 行表示每头奶牛的能力指数 \(p(1),p(2)...p(N)\),接下来的 \(N?1\) 行描述每头奶牛 \((2\) ~ \(N)\) 的上司(父节点)的编号。

【输出】

输出包括 \(N\) 行,输出的第 \(i\) 行表示奶牛 \(i\) 的下属中能力比奶牛 \(i\) 高的数量。

【样例】

样例输入:
5
804289384
846930887
681692778
714636916
957747794
1
1
2
3

样例输出:
2
0
1
0
0

【数据规模】

\(100\%\) \(1 \leqslant N \leqslant 1e5,1 \leqslant p(i) \leqslant 1e9\)

【分析】

问题实质:求树上逆序对

解法一: 树状数组 / 线段树 + \(dfs\)

对能力值开一个权值线段树\((\)范围较大 \(1e9\),加一个离散化\()\)。
对于每个节点 \(i\),在线段树中查询数值大于 \(a[i]\) 的数的个数,再对于点 \(i\) 所有儿子节点的权值都进行一次【单点修改】操作,将它们都加入进去,再一次进行同样的查询,用其减去前面查询的结果就得到变化量 ?,即刚刚加入的这个儿子所统领的树\((\)包括它自己\()\)中能力大于 \(p(i)\) 的牛的数量,即 \(ans[i]\)。树状数组可用同样的思路。

\(【Code】\)

(线段树)

#include<algorithm>
#include<cstdio>
#define pl p<<1//左段儿子的编号
#define pr p<<1|1//右段儿子的编号
#define Re register int
#define F(a,b) for(i=a;i<=b;++i)
const int N=1e5+3;
int x,i,n,t,Q[N],ip[N],ans[N],nex[N],head[N],tree[N<<2];
struct A{int x,i;bool operator<(A b)const{return x<b.x;};}a[N];
//【重载"<"运算符】自己动手,丰衣足食...
inline void add(Re x,Re y){Q[++t]=y,nex[t]=head[x],head[x]=t;}//数组维护一个链表
inline void in(Re &x){//【快读】自己动手,丰衣足食...
    x=0;char c=getchar();
    while(c<'0'||c>'9')c=getchar();
    while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+(c^48),c=getchar();
}
inline void change(Re p,Re x,Re L,Re R){
//【单点修改】每个人仅有一个能力值,所以是单点修改
    if(L==R){++tree[p];return;}//单点修改的边界:L==R
    Re mid=L+R>>1;
    if(x<=mid)change(pl,x,L,mid);//注意:单点修改中,进入递归函数的前提:x<=mid
    else change(pr,x,mid+1,R);//x>mid
    tree[p]=tree[pl]+tree[pr];//递归完子树后更新
}
inline int ask(Re p,Re l,Re r,Re L,Re R){//【区间查询】
    if(l<=L&&R<=r)return tree[p];//区间查询的边界: l<=L&&R<=r
    Re ans=0,mid=L+R>>1;
    if(l<=mid)ans+=ask(pl,l,r,L,mid);//注意区间查询中,进入递归函数的前提:l<=mid
    if(r>mid)ans+=ask(pr,l,r,mid+1,R);//r>mid
    return ans;
}
inline void dfs(Re x){//搜索遍历树
    ans[x]-=ask(1,ip[x]+1,n,1,n);//这里计算出的是其他节点的数据,要减去
    for(Re i=head[x];i;i=nex[i])dfs(Q[i]);//先让儿子更新线段树【单点修改】
    ans[x]+=ask(1,ip[x]+1,n,1,n);
    //查询大于当前这个点x的等级的一共有多少个【区间查询】,再减去原先的
    change(1,ip[x],1,n);//让当前这个点x的等级进行【单点修改】
}
int main(){
    in(n);
    F(1,n)in(a[i].x),a[i].i=i;
    std::sort(a+1,a+n+1);//【离散化】依据每个人的能力p(i)排出他们的等级
    F(1,n)ip[a[i].i]=i;//用ip[]记录下每个点的等级
    F(2,n)in(x),add(x,i);//链表建树,实现从根节点到儿子叶节点的遍历
    dfs(1);//从没有上司的根节点1开始遍历
    F(1,n)printf("%d\n",ans[i]);
}

(树状数组)

#include<algorithm>
#include<cstdio>
#define Re register int
#define F(a,b) for(i=a;i<=b;++i)
const int N=1e5+3;
int x,i,n,t,C[N],Q[N],ip[N],ans[N],nex[N],head[N];
struct A{int x,i;bool operator<(A b)const{return x<b.x;};}a[N];
inline void add(Re x,Re y){Q[++t]=y,nex[t]=head[x],head[x]=t;}
inline void in(Re &x){
    x=0;char c=getchar();
    while(c<'0'||c>'9')c=getchar();
    while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+(c^48),c=getchar();
}
inline void addx(Re x){while(x<=n)++C[x],x+=x&(-x);}
inline int ask(Re x){int ans=0;while(x)ans+=C[x],x-=x&(-x);return ans;}
inline void dfs(Re x){
    ans[x]-=ask(n)-ask(ip[x]);
    for(Re i=head[x];i;i=nex[i])dfs(Q[i]);
    ans[x]+=ask(n)-ask(ip[x]);
    addx(ip[x]);
}
int main(){
    in(n);
    F(1,n)in(a[i].x),a[i].i=i;
    std::sort(a+1,a+n+1);
    F(1,n)ip[a[i].i]=i;
    F(2,n)in(x),add(x,i);
    dfs(1);
    F(1,n)printf("%d\n",ans[i]);
}

解法二:线段树合并

对于 \(N\) 个点每个点都开一个权值线段树。
对公司上司下属关系形成的树进行\(dfs\)遍历,对于每个节点 \(i\) ,将它所有儿子的权值线段树合并到 \(tree[pt[i]](\) 用 \(pt[i]\) 表示节点 \(i\) 所对应的权值线段树的根节点编号\()\),然后在 \(tree[pt[i]]\) 中查询数值大于 \(a[i]\) 的数的个数即为 \(ans[i]\)。

\(【Code】\)

#include<algorithm>
#include<cstdio>
#define pl tr[p].lp//左段儿子的编号
#define pr tr[p].rp//右段儿子的编号
#define Re register int
#define F(a,b) for(i=a;i<=b;++i)
const int N=1e5+3;
int x,i,n,t,cnt,Q[N],pt[N],ip[N],ans[N],nex[N],head[N];
struct QAQ{int g,lp,rp,L,R;}tr[N*18];//这里的空间消耗要注意,开太小死活过不了
struct A{int x,i;bool operator<(A b)const{return x<b.x;};}a[N];
//【重载"<"运算符】自己动手,丰衣足食...
inline void add(Re x,Re y){Q[++t]=y,nex[t]=head[x],head[x]=t;}//数组维护一个链表
inline void in(Re &x){//【快读】自己动手,丰衣足食...
    x=0;char c=getchar();
    while(c<'0'||c>'9')c=getchar();
    while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+(c^48),c=getchar();
}
inline void change(Re p,Re x){//【单点修改】(虽然在这道题中没什么用)
    Re L=tr[p].L,R=tr[p].R;//把编号为p的这个区间所覆盖的范围L,R提出来
    if(L==R){++tr[p].g;return;}//单修的递归边界:L==R
    Re mid=L+R>>1;
    if(x<=mid)change(pl,x);//注意:单修中进入递归函数的前提:x<=mid
    else change(pr,x);//x>mid
    tr[p].g=tr[pl].g+tr[pr].g;//递归完子树后更新
}
inline void creat(Re &p,Re x,Re L,Re R){//建立线段树
    tr[p=++cnt].L=L,tr[p].R=R;//动态开点,并记录新编号子树所覆盖的区间范围L,R
    if(L==R){tr[p].g=1;return;}//建立线段树时是一个点一个点的建,所以边界类似于单修
    Re mid=L+R>>1;
    if(x<=mid)creat(pl,x,L,mid);//递归进入前提也类似于单修
    else creat(pr,x,mid+1,R);
    tr[p].g=tr[pl].g+tr[pr].g;//递归完子树后更新
}
inline int merge(Re p,Re q){//【线段树合并】
    if(!p)return q;if(!q)return p;
    //当需要合并的点的其中一个编号为0时 (即为空),返回另一个编号
    tr[p].g+=tr[q].g,p;//把q合并到p上面去
    pl=merge(pl,tr[q].lp);//分别合并左子树,右子树
    pr=merge(pr,tr[q].rp);
    return p;
}
inline int ask(Re p,Re l,Re r){//【区间查询】
    if(!p)return 0;
    Re L=tr[p].L,R=tr[p].R;
    if(l<=L&&R<=r)return tr[p].g;//区间查询的边界: l<=L&&R<=r
    Re ans=0,mid=L+R>>1;
    if(l<=mid)ans+=ask(pl,l,r);//注意区间查询中,进入递归函数的前提:l<=mid
    if(r>mid)ans+=ask(pr,l,r);//r>mid
    return ans;
}
inline void dfs(Re x){//搜索遍历树
    for(Re i=head[x];i;i=nex[i]){
        dfs(Q[i]);//先完善儿子的线段树
        merge(pt[x],pt[Q[i]]);//把儿子合并进来
    }
    ans[x]=ask(pt[x],ip[x]+1,n);
    //查询大于当前这个点x的等级的一共有多少个【区间查询】,再减去原先的
}
int main(){
    in(n);
    F(1,n)in(a[i].x),a[i].i=i;
    std::sort(a+1,a+n+1);//【离散化】依据每个人的能力p(i)排出他们的等级
    F(1,n)ip[a[i].i]=i,creat(pt[a[i].i],i,1,n);
    //用ip[]记录下每个点的等级,并建立线段树
    //【注意】:进入建树时的编号是原本输入时的编号即a[i].i
    F(2,n)in(x),add(x,i);//链表建搜索树,实现从根节点到儿子叶节点的遍历
    dfs(1);//从没有上司的根节点1开始遍历
    F(1,n)printf("%d\n",ans[i]);
}


【题解】晋升者计数 Promotion Counting [USACO 17 JAN] [P3605]

原文地址:https://www.cnblogs.com/Xing-Ling/p/10803233.html

时间: 2024-10-08 08:59:20

【题解】晋升者计数 Promotion Counting [USACO 17 JAN] [P3605]的相关文章

usaco 17.Jan 铜组T3

上午在打usaco月赛的铜组题,T1T2是用来秒杀的,然而T3卡了一上午,下面给出题面: 题意大概就是输入一个N*N的矩阵,矩阵中元素只有0与1两种状态,每次操作以左上角的点为矩阵中某一矩阵的左上方顶点,将该矩阵中所有元素状态改变(即0变为1,1变为0),求将矩阵中元素全部变为0的最小次数. 第一次看到样例的时候以为就是一道的DFS或BFS的搜索题,然后果断写了DFS,很正常就WA了.然而其实这道题需要用到贪心... 题目中提到,一个矩阵被改变两次之后还是原先的状态,那既然这样,如果将右下角的点

洛谷P3605 [USACO17JAN]Promotion Counting晋升者计数

洛谷P3605 [USACO17JAN]Promotion Counting晋升者计数 题目描述 奶牛们又一次试图创建一家创业公司,还是没有从过去的经验中吸取教训--牛是可怕的管理者! 为了方便,把奶牛从 $1 \cdots N(1 \leq N \leq 100, 000)$ 编号,把公司组织成一棵树,1 号奶牛作为总裁(这棵树的根节点).除了总裁以外的每头奶牛都有一个单独的上司(它在树上的 “双亲结点”).所有的第 $i$ 头牛都有一个不同的能力指数 $p(i)$,描述了她对其工作的擅长程度

[BZOJ4756][Usaco2017 Jan]Promotion Counting 树状数组

4756: [Usaco2017 Jan]Promotion Counting Time Limit: 10 Sec  Memory Limit: 128 MBSubmit: 305  Solved: 217[Submit][Status][Discuss] Description The cows have once again tried to form a startup company, failing to remember from past experience t hat cow

洛谷——P1596 [USACO10OCT]湖计数Lake Counting

P1596 [USACO10OCT]湖计数Lake Counting 题目描述 Due to recent rains, water has pooled in various places in Farmer John's field, which is represented by a rectangle of N x M (1 <= N <= 100; 1 <= M <= 100) squares. Each square contains either water ('W'

经典排序算法 - 计数排序Counting sort

经典排序算法 - 计数排序Counting sort 注意与基数排序区分,这是两个不同的排序 计数排序的过程类似小学选班干部的过程,如某某人10票,作者9票,那某某人是班长,作者是副班长 大体分两部分,第一部分是拉选票和投票,第二部分是根据你的票数入桶 看下具体的过程,一共需要三个数组,分别是待排数组,票箱数组,和桶数组 var unsorted = new int[] { 6, 2, 4, 1, 5, 9 };  //待排数组 var ballot = new int[unsorted.Len

USACO翻译:USACO 2012 JAN三题(1)

USACO 2012 JAN(题目一) 一.题目概览 中文题目名称 礼物 配送路线 游戏组合技 英文题目名称 gifts delivery combos 可执行文件名 gifts delivery combos 输入文件名 gifts.in delivery.in combos.in 输出文件名 gifts.out delivery.out combos.out 每个测试点时限 1秒 1秒 1秒 测试点数目 10 10 10 每个测试点分值 10 10 10 比较方式 全文比较 全文比较 全文比

USACO翻译:USACO 2013 JAN三题(1)

USACO 2013 JAN 一.题目概览 中文题目名称 镜子 栅栏油漆 奶牛排队 英文题目名称 mirrors paint lineup 可执行文件名 mirrors paint lineup 输入文件名 mirrors.in paint.in lineup.in 输出文件名 mirrors.out paint.out lineup.out 每个测试点时限 1秒 1秒 1秒 测试点数目 10 10 10 每个测试点分值 10 10 10 比较方式 全文比较 全文比较 全文比较 二.运行内存限制

USACO翻译:USACO 2014 JAN三题(2)

USACO 2014 JAN 一.题目概览 中文题目名称 队伍平衡 滑雪录像 滑雪场建设 英文题目名称 bteams recording skicourse 可执行文件名 bteams recording skicourse 输入文件名 bteams.in recording.in skicourse.in 输出文件名 bteams.out recording.out skicourse.out 每个测试点时限 1秒 1秒 1秒 测试点数目 10 10 10 每个测试点分值 10 10 10 比

USACO翻译:USACO 2012 JAN三题(2)

USACO 2012 JAN(题目二) 一.题目概览 中文题目名称 叠干草 分干草 奶牛跑步 英文题目名称 stacking baleshare cowrun 可执行文件名 stacking baleshare cowrun 输入文件名 stacking.in baleshare.in cowrun.in 输出文件名 stacking.out baleshare.out cowrun.out 每个测试点时限 1秒 1秒 1秒 测试点数目 10 10 10 每个测试点分值 10 10 10 比较方