ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

PAT——Stack(使用树状数组避免运行超时)

2021-08-04 19:01:21  阅读:146  来源: 互联网

标签:tmp PAT cout 树状 int mid 超时 Stack left


Stack

题目

https://pintia.cn/problem-sets/994805342720868352/problems/994805417945710592

AC代码

#include<bits/stdc++.h>
#define lowbit(i) (i&(-i))
using namespace std;
const int inf=0xffffff,maxn=100005;
stack<int> s;
int c[maxn];
void update(int x,int v)
{
	for(int i=x;i<maxn;i+=lowbit(i))
	{
		c[i]+=v;
	}
}
int getsum(int x)
{
	int sum=0;
	for(int i=x;i>=1;i-=lowbit(i))
	{
		sum+=c[i];
	}
	return sum;
}
void PeekMid()
{
	int left=1,right=maxn,k=(s.size()+1)/2;
	while(left<right)
	{
		int mid=(left+right)/2;
		if(getsum(mid)>=k) right=mid;
		else left=mid+1;
	}
	cout<<left<<endl;
}
int main()
{
	int n,tmp;
	cin>>n;
	string str;
	while(n--)
	{
		cin>>str;
		if(str=="Pop")
		{
			if(s.empty()) cout<<"Invalid"<<endl;
			else 
			{
				tmp=s.top();
				s.pop();
				update(tmp,-1);
				cout<<tmp<<endl;
			}
		}
		else if(str=="Push")
		{
			cin>>tmp;
			s.push(tmp);
			update(tmp,1);
		}
		else
		{
			if(s.empty()) cout<<"Invalid"<<endl;
			else PeekMid();
		}
	}
}

参考

PAT1057 Stack (30)(树状数组)

说明

本题使用了树状数组,不了解相关知识点的同学可以看这篇文章——c /c++ 树状数组 入门

首先要明确c[i]表示前i个数中实际出现了几个数

关于PeekMid函数,在理解时要注意,假设c[7]=3,c[8]=3,c[9]=3,函数找到的应该是下标7,因为上面三组分别表示前7、8、9个数中实际出现了3个数,因此可以推导出8、9两个数字是没有出现的,只有7出现了,所以7就是我们要找的答案

标签:tmp,PAT,cout,树状,int,mid,超时,Stack,left
来源: https://blog.csdn.net/ljhsq/article/details/119390701

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有