2016华为机试-自动售货机

#include<iostream>
#include<string>
#include<vector>
using namespace std;
int main()
{
	string str;
	while(getline(cin,str))
	{
		int countA1=5;
		int countA2=10;
		int countA3=14;
		int money1=5;
		int money2=5;
		int money3=3;
		vector<int> givemoney;
		int givesum=0;
		vector<int> backmoney;
		vector<int>buyinfo;
		int buysum=0;
		for (int i = 0; i < str.size(); ++i)
		{
			if (str[i]=='p')
			{
				i=i+2;
				if (str[i]=='5')
				{
					money3++;
				}
				if (str[i]=='2')
				{
					money2++;
				}
				if(str[i]=='1')
				{
					money1++;
				}
				givemoney.push_back(str[i]-'0');
				givesum+=str[i]-'0';
				continue;
			}
			if (str[i]=='b')
			{
				string temp;
				temp+=str[i+2];
				temp+=str[i+3];
				i+=3;
				if (temp=="A1")
				{
					countA1--;
					buysum+=2;
					temp="";
					continue;
				}
				if (temp=="A2")
				{
					countA2--;
					buysum+=3;
					temp="";
					continue;
				}
				if (temp=="A3")
				{
					countA3--;
					buysum+=4;
					temp="";
					continue;
				}
			}
		}
		int leavesum=givesum-buysum;
		if (buysum==0)
		{
			for (int t = 0; t <givemoney.size(); ++t)
			{
				if (givemoney[t]==5)
				{
					money3--;
				}
				if (givemoney[t]==2)
				{
					money2--;
				}
				if (givemoney[t]==1)
				{
					money1;
				}
			}
		}
		while(leavesum>0&&buysum!=0)
		{
			if (leavesum>=5)
		    {
			    int c5=leavesum/5;
			    money3-=c5;
			    leavesum=leavesum%5;
			    continue;
		   }
		   if (leavesum>=2)
		   {
		   		int c2=leavesum/2;
		   		money2-=c2;
		   		leavesum=leavesum%2;
		   		continue;
		   }
		   leavesum-=1;
		   money1-=1;
		}
		cout<<"A1"<<" "<<countA1<<","<<"A2"<<" "<<countA2<<","<<"A3"<<" "<<countA3<<";";
		cout<<1<<" "<<money1<<","<<2<<" "<<money2<<","<<5<<" "<<money3<<endl;
	}
	return 0;
}