nyoj8 一種排序（用set）

一種排序

時間限制：3000 ms  |  內存限制：65535 KB

難度：3

描述

現在有很多長方形，每一個長方形都有一個編號，這個編號可以重複；還知道這個長方形的寬和長，編號、長、寬都是整數；現在要求按照一下方式排序（默認排序規則都是從小到大）；

1.按照編號從小到大排序

2.對於編號相等的長方形，按照長方形的長排序；

3.如果編號和長都相同，按照長方形的寬排序；

4.如果編號、長、寬都相同，就只保留一個長方形用於排序,刪除多餘的長方形；最後排好序按照指定格式顯示所有的長方形；

輸入

第一行有一個整數 0<n<10000,表示接下來有n組測試數據；

每一組第一行有一個整數 0<m<1000，表示有m個長方形；

接下來的m行，每一行有三個數 ，第一個數表示長方形的編號，

第二個和第三個數值大的表示長，數值小的表示寬，相等

說明這是一個正方形（數據約定長寬與編號都小於10000）；

輸出

順序輸出每組數據的所有符合條件的長方形的 編號 長 寬

樣例輸入

1

8

1 1 1

1 1 1

1 2 1

1 1 2

1 2 2

2 2 1

2 1 1

2 1 2

樣例輸出

1 1 1

1 2 1

2 1 1

1 2 2

2 2 1

#include <iostream>
#include <cstdio>
#include <set>
#include <iterator>
using namespace std;

struct node {
    int num, length, width;
};
bool operator < (const node &a, const node &b) {
    if(a.num != b.num)
        return a.num < b.num;
    if(a.length != b.length)
        return a.length < b.length;
    return a.width < b.width;
}
ostream & operator << (ostream &out, const node &a) {
    return out << a.num << " " << a.length << " " << a.width;
}
set<node> s;
int main() {
    int n, m, a, b;
    node t;
    scanf("%d", &n);
    while(n--) {
        scanf("%d", &m);
        s.clear();
        while(m--) {
            scanf("%d%d%d", &t.num, &a, &b);
            t.length = a >= b ? a : b;
            t.width = a <= b ? a : b;
            s.insert(t);
        }
        //set<node>::iterator i;
        //for(i = s.begin(); i != s.end(); ++i) {
           //printf("%d %d %d\n", (*i).num, (*i).length, (*i).width);
        //}
       copy(s.begin(), s.end(), ostream_iterator<node>(cout,"\n"));
    }
    return 0;
}

兩種輸出方式都可以，也可以在重載 << 時加上換行，這樣，ostream_iterator<node>()可以只有一個參數cout，即：

ostream & operator << (ostream &out, const node &a) {
    return out << a.num << " " << a.length << " " << a.width << endl;
}

copy(s.begin(), s.end(), ostream_iterator<node>(cout));

關於copy函數的解釋：http://blog.csdn.net/jingweihu/article/details/23241269

使用ostream_iterator< >()時，記得引用頭文件#include <iterator>

當然輸入可以通過重載 >>運算符實現，而且可以更快，然而寶寶懶~~

istream& operator >> (istream& in, node &t) {
    in >> t.num;
    int a, b;
    cin >> a >> b;
    t.length = max(a, b);
    t.width = min(a, b);
    return in;
}