【牛客】复数集合（C++）

题目描述

    一个复数（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 <iostream>
#include <cstdio>
#include <queue>
#include <string>

using namespace std;

struct complex
{
    int real;  //实部
    int imag;  //虚部
    complex(int a,int b): real(a),imag(b){}
    bool operator<(complex c)const    //结构体内运算符重载
    {
        return real*real+imag*imag<c.real*c.real+c.imag*c.imag;
    }
};

int main()
{
    int n;
    scanf("%d",&n);
    priority_queue<complex> myqueue;  //优先级队列
    while(n)
    {
        string op;
        getline(cin,op);
        if(op=="Pop")
        {
            if(myqueue.empty())
            {
                printf("empty\n");
            }
            else
            {
                complex temp;
                temp=myqueue.top();
                myqueue.pop();
                printf("%d+i%d\n",temp.real,temp.imag);
                printf("SIZE=%d\n",myqueue.size());
            }
        }
        else
        {
            int a,b;
            scanf("%d+i%d",&a,&b);
            myqueue.push(complex(a,b));
            printf("SIZE=%d\n",myqueue.size());
        }
        n--;
    }
    return 0;
}