排队

排队

n 个小朋友排成一排,从左到右依次编号为 1n

i 个小朋友的身高为 hi

虽然队伍已经排好,但是小朋友们对此并不完全满意。

对于一个小朋友来说,如果存在其他小朋友身高比他更矮,却站在他右侧的情况,该小朋友就会感到不满。

每个小朋友的不满程度都可以量化计算,具体来说,对于第 i 个小朋友:

  • 如果存在比他更矮且在他右侧的小朋友,那么他的不满值等于其中最靠右的那个小朋友与他之间的小朋友数量。
  • 如果不存在比他更矮且在他右侧的小朋友,那么他的不满值为 1

请你计算并输出每个小朋友的不满值。

注意,第 1 个小朋友和第 2 个小朋友之间的小朋友数量为 0,第 1 个小朋友和第 4 个小朋友之间的小朋友数量为 2

输入格式

第一行包含整数 n

第二行包含 n 个整数 h1,h2,,hn

输出格式

共一行,输出 n 个整数,第 i 个整数为第 i 个小朋友的不满值。

数据范围

5 个测试点满足 2n5
所有测试点满足 2n1051hi109

输入样例1:

6
10 8 5 3 50 45

输出样例1:

2 1 0 -1 0 -1

输入样例2:

7
10 4 6 3 2 8 15

输出样例2:

4 2 1 0 -1 -1 -1 

输入样例3:

5
10 3 1 10 11

输出样例3:

1 0 -1 -1 -1 

 

解题思路

  这题就是就是考察在每个数右边且严格比它小的数中,找到下标最大的那个数。这个就是单调栈的模型,可以参考我之前写过的博客:https://www.cnblogs.com/onlyblues/p/16736106.html。

  不过上面那个博客考虑的情况都是在某个数的左边去找,这题就反过来在右边找。本质上是一样的,我们只要反过来枚举,对于i,j(i<j),如果val>aiaj,那么ai永远不会作为答案输出,就没有存在必要了,因此整个栈是单调的:

  因此可以二分出小于val的最大数,栈就存着这个数的下标。

  AC代码如下:

复制代码
 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 
 4 const int N = 1e5 + 10;
 5 
 6 int a[N];
 7 int stk[N], tp;
 8 int ans[N];
 9 
10 int main() {
11     int n;
12     scanf("%d", &n);
13     for (int i = 0; i < n; i++) {
14         scanf("%d", a + i);
15     }
16     memset(ans, -1, sizeof(ans));
17     for (int i = n - 1; i >= 0; i--) {
18         int l = 1, r = tp;
19         while (l < r) {
20             int mid = l + r >> 1;
21             if (a[stk[mid]] < a[i]) r = mid;
22             else l = mid + 1;
23         }
24         if (tp && a[stk[l]] < a[i]) ans[i] = stk[l] - i - 1;
25         if (!tp || a[i] < a[stk[tp]]) stk[++tp] = i;    // 因为整个栈内的元素式单调递减的,因此只有a[i]小于栈顶元素才可以压入栈
26     }
27     for (int i = 0; i < n; i++) {
28         printf("%d ", ans[i]);
29     }
30     
31     return 0;
32 }
复制代码

  再贴个树状数组的代码,树状数组维护的是值域,每个值的最大下标。

  AC代码如下:

复制代码
 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 
 4 const int N = 1e5 + 10;
 5 
 6 int a[N];
 7 int xs[N], sz;
 8 int tr[N];
 9 int ans[N];
10 
11 int find(int x) {
12     int l = 1, r = sz;
13     while (l < r) {
14         int mid = l + r >> 1;
15         if (xs[mid] >= x) r = mid;
16         else l = mid + 1;
17     }
18     return l;
19 }
20 
21 int lowbit(int x) {
22     return x & -x;
23 }
24 
25 void add(int x, int c) {
26     for (int i = x; i <= sz; i += lowbit(i)) {
27         tr[i] = max(tr[i], c);
28     }
29 }
30 
31 int query(int x) {
32     int ret = 0;
33     for (int i = x; i; i -= lowbit(i)) {
34         ret = max(ret, tr[i]);
35     }
36     return ret;
37 }
38 
39 int main() {
40     int n;
41     scanf("%d", &n);
42     for (int i = 1; i <= n; i++) {
43         scanf("%d", a + i);
44         xs[++sz] = a[i];
45     }
46     sort(xs + 1, xs + sz + 1);
47     sz = unique(xs + 1, xs + sz + 1) - xs - 1;
48     memset(ans, -1, sizeof(ans));
49     for (int i = n; i; i--) {
50         int t = query(find(a[i]) - 1);
51         if (t) ans[i] = t - i - 1;
52         add(find(a[i]), i);
53     }
54     for (int i = 1; i <= n; i++) {
55         printf("%d ", ans[i]);
56     }
57     
58     return 0;
59 }
复制代码

 

参考资料

  AcWing 4721. 排队(AcWing杯 - 周赛):https://www.acwing.com/video/4540/

posted @   onlyblues  阅读(168)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 单线程的Redis速度为什么快?
· 展开说说关于C#中ORM框架的用法!
· Pantheons:用 TypeScript 打造主流大模型对话的一站式集成库
· SQL Server 2025 AI相关能力初探
· 为什么 退出登录 或 修改密码 无法使 token 失效
Web Analytics
点击右上角即可分享
微信分享提示