复数集合

题目描述
一个复数（x+iy）集合，两种操作作用在该集合上： 1、Pop 表示读出集合中复数模值最大的那个复数，如集合为空 输出 empty ，不为空就输出最大的那个复数并且从集合中删除那个复数，再输出集合的大小SIZE； 2 Insert a+ib 指令（a，b表示实部和虚部），将a+ib加入到集合中 ，输出集合的大小SIZE； 最开始要读入一个int n，表示接下来的n行每一行都是一条命令。

输入描述:
输入有多组数据。
每组输入一个n(1<=n<=1000)，然后再输入n条指令。

输出描述:
根据指令输出结果。
模相等的输出b较小的复数。
a和b都是非负数。

示例1
输入
3
Pop
Insert 1+i2
Pop
输出
empty
SIZE = 1
1+i2
SIZE = 0

题目解析：因为需要比较大小，所以需要将实部和虚部提取出来，计算模大小。

代码：

#include<stdio.h>
#include<math.h>
#include<algorithm>
#include<string.h>
#include<iostream>
#include<iomanip>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<queue>
using namespace std;
//复数的模， (实数的平方 + 虚数的平方) 的开方
//不用考虑输入，标准输入，有a，b 还有+ 
//仍然有问题，虽然可以通过，但是我没有考虑模相等b的情况，自己完善
 
int computeNumber(string a){    //分割值 
	int count = a.size() - 1;
	int number = 0;
	for(int i = 0 ; i < a.size(); i++){
		number += (a[i] - '0') * pow(10,count);   //卡了半个小时，忘了转成数字 
		count--;
	}
	return number;
}

long long computeModel(string str){  //输出模 
	int index;
	index = str.find("+");
	string a = str.substr(0,index);
	string b = str.substr(index + 2);
	long long number1 = computeNumber(a);
	long long number2 = computeNumber(b);
	long long rst = sqrt(number1 * number1 + number2 * number2);
	//cout <<"a : " << a << " b : " << b <<" number : " << number1 << " number : " << number2 << " rst : " << rst << endl;
	return rst;
}

bool cmp(pair<string,long long> p1 ,pair<string,long long> p2){    //按模的大小排序 
	return p1.second < p2.second;
}

int main()
{
	int n;
	vector<pair<string,long long>> vc;
	string flag , str;
	while(cin >> n){
		for(int i = 0 ; i < n; i++){
			cin >> flag;
			if(flag == "Pop"){
				if(vc.empty()){
					cout << "empty" << endl;					
				}else{
					cout << vc[vc.size() - 1].first << endl;
					vc.pop_back();
					cout <<"SIZE = " << vc.size() << endl;
				}
			}else if(flag == "Insert"){
				cin >> str;
				long long model = computeModel(str);
				vc.push_back(make_pair(str,model));
				sort(vc.begin() , vc.end() , cmp);    //按模大小排序 
				cout << "SIZE = " << vc.size() << endl;
			}
		}
		vc.clear();
	}
    return 0;
}