PAT-1057. Stack (30)--树状数组

// 1057.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include<string>
#include<stack>
using namespace std;

#define MAX 100010

stack<int> s;

int tree[MAX]={0};

int lowbit(int ind)
{
	return ind & -ind;
}

void update(int ind,int val)
{
	while(ind<=MAX)
	{
		tree[ind]+=val;
		ind+=lowbit(ind);
	}
}

int getsum(int ind)
{
	int sum=0;
	while(ind>0) //not >=0
	{
		sum+=tree[ind];
		ind-=lowbit(ind);
	}
	return sum;
}

int findmid(int number,int l,int r)
{
	if(l==r)
		return l;
	int mid=(l+r)/2;
	int sum=getsum(mid);
	if(sum<number)
	{
		return findmid(number,mid+1,r);
	}
	else
	{
		return findmid(number,l,mid);  
	}
	/*
	  最后return mid会出错
 	*/
}

int main()
{
	int num;
	string op;
	char op_name[15];
	int tmp;
	freopen("1057.txt","r",stdin);
	while(scanf("%d",&num)!=EOF)
	{
       for(int i=0;i<num;i++)
	   {
		   scanf("%s",op_name);
		   op=string(op_name);
		   if(op=="Pop")
		   {
			   if(!s.empty())
			   {
				   tmp=s.top();
				   update(tmp,-1);
				   printf("%d\n",tmp);
				   s.pop();
			   }
			   else
			   {
				   printf("Invalid\n");
			   }
		   }
		   if(op=="Push")
		   {
			   scanf("%d",&tmp);
			   s.push(tmp);
			   update(tmp,1);
		   }
		   if(op=="PeekMedian")
		   {
			   int size=s.size();
			   if(size==0)
			   {
				   printf("Invalid\n");
				   continue;
			   }
			   if(size%2==0)
				   size=size/2;
			   else
				   size=(size+1)/2;

			   int mid=findmid(size,1,MAX);

			   printf("%d\n",mid);
		   }
	   }
	}
	return 0;
}