堆栈中的剩余数字C++


向一个空栈中依次存入正整数, 假设入栈元素n(1<=n<=2^31-1)按顺序依次为nx...n4、n3、n2、n1, 每当元素入栈时,如果n1=n2+...+ny(y的范围[2,x],1<=x<=1000),则n1~ny全部元素出栈,重新入栈新元素m(m=2*n1)。

如:依次向栈存入6、1、2、3, 当存入6、1、2时,栈底至栈顶依次为[6、1、2];当存入3时,3=2+1,3、2、1全部出栈,重新入栈元素6(6=2*3),此时栈中有元素6;因为6=6,所以两个6全部出栈,存入12,最终栈中只剩一个元素12。

输入描述: 使用单个空格隔开的正整数的字符串,如"5 6 7 8", 左边的数字先入栈,输入的正整数个数为x,1<=x<=1000。

输出描述: 最终栈中存留的元素值,元素值使用空格隔开,如"8 7 6 5", 栈顶数字在左边。

示例1 输入 5 10 20 50 85 1 输出 1 170 说明 5+10+20+50=85, 输入85时,5、10、20、50、85全部出栈,入栈170,最终依次出栈的数字为1和170。

#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
using namespace std;;
int main() 
{
	vector<int> a; int len; char c; int sum = 0; a.push_back(0);
	while(scanf("%d",&len))
	{
		a.push_back(len);
		c = getchar();
		if (c == '\n')break;
	}
	len = a.size();
	for(int i=1,j=0;i<len;++i)//i表示入栈数字,j为栈顶;
	{
		sum = 0;
		for(int k=j;k>=0;--k)
		{
			sum += a[k];
			if(sum==a[i])
			{
				a[k] = 2 * a[i]; a[i] = 0;
				for (int m = k + 1; m <= j; ++m)a[m] = 0;
				j = k+1;sum=INT_MIN;
				break;
			}
			if (sum > a[i])break;
		}
		if(sum!=INT_MIN)
		{
			 a[j] = a[i]; a[i] = 0;j++;
		}
	}
	for (int i = len-1; i > 0; --i)if (a[i] != 0)cout << a[i] << " ";cout<<a[0];
	return 0;
}

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM