排队
排队
个小朋友排成一排,从左到右依次编号为 。
第 个小朋友的身高为 。
虽然队伍已经排好,但是小朋友们对此并不完全满意。
对于一个小朋友来说,如果存在其他小朋友身高比他更矮,却站在他右侧的情况,该小朋友就会感到不满。
每个小朋友的不满程度都可以量化计算,具体来说,对于第 个小朋友:
- 如果存在比他更矮且在他右侧的小朋友,那么他的不满值等于其中最靠右的那个小朋友与他之间的小朋友数量。
- 如果不存在比他更矮且在他右侧的小朋友,那么他的不满值为 。
请你计算并输出每个小朋友的不满值。
注意,第 个小朋友和第 个小朋友之间的小朋友数量为 ,第 个小朋友和第 个小朋友之间的小朋友数量为 。
输入格式
第一行包含整数 。
第二行包含 个整数 。
输出格式
共一行,输出 个整数,第 个整数为第 个小朋友的不满值。
数据范围
前 个测试点满足 。
所有测试点满足 ,。
输入样例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。
不过上面那个博客考虑的情况都是在某个数的左边去找,这题就反过来在右边找。本质上是一样的,我们只要反过来枚举,对于,如果,那么永远不会作为答案输出,就没有存在必要了,因此整个栈是单调的:
因此可以二分出小于的最大数,栈就存着这个数的下标。
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/
本文来自博客园,作者:onlyblues,转载请注明原文链接:https://www.cnblogs.com/onlyblues/p/16907945.html
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 单线程的Redis速度为什么快?
· 展开说说关于C#中ORM框架的用法!
· Pantheons:用 TypeScript 打造主流大模型对话的一站式集成库
· SQL Server 2025 AI相关能力初探
· 为什么 退出登录 或 修改密码 无法使 token 失效