复数集合（北邮复试上机）

前言：

21考研，记录一下准备复试路上写下的垃圾代码。本来啃《算法笔记》，但是感觉太多了做不完，改做王道机试指南。

题目描述：

一个复数（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都是非负数。

解答

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<cmath>
#include<string>
#include<queue>
using namespace std;

struct plu {
	int real;
	int img;
	plu(int a, int b) :real(a), img(b) {}
	bool operator< (plu c) const {
		if (real * real + img * img == c.real * c.real + c.img * c.img) {
			return img > c.img;
		}
		else
			return real * real + img * img < c.real * c.real + c.img * c.img;
	}
};
int main()
{
	int n;
	string op;
	string add = "+";
	string i = "i";
	while (cin >> n) {
		priority_queue<plu> Q;
		string str;
		while (n--) {
			cin >> op;
			if (op == "Pop") {
				if (Q.empty())
					printf("empty\n");
				else {
					plu cur = Q.top();
					printf("%d+i%d\n", cur.real, cur.img);
					Q.pop();
					printf("SIZE = %d\n", Q.size());
				}
			}
			else {
				cin >> str;
				int real;
				int img;
				real = stoi(str.substr(0, str.find(add)));
				img = stoi(str.substr(str.find(i) + 1, str.length() - str.find(i) - 1));
				plu cur(real, img);
				Q.push(cur);
				printf("SIZE = %d\n", Q.size());
			}
		}
	}
	return 0;
}