使用compareTo排序对象(多种类型)

时间:2013-07-08 21:09:08

标签: java compareto

首先,我有一个名为新闻文章的对象,它有三个我需要排序的属性:

Year(int),Month(int),type(String - online,paper)

例如,它会是这样的:

  • Online 2013 4
  • 在线2013 1
  • Online 2009 11
  • Online 2008 4
  • Paper 2012 12
  • Paper 2011 9

发生的事情是月份和年份似乎正确排序,但我遇到按类型排序的问题。在compareTo中按字符串排序的正确方法是什么?

目前的结果:

  • Paper 2012 12
  • Paper 2011 9
  • Online 2013 4
  • 在线2013 1
  • Online 2009 11
  • Online 2008 4

这是我的方法(我道歉它有点古怪,我一直在尝试不同的方式按类型排序并进行实验):

@Override
public int compareTo(Article rs) {

    Integer x = 0;


        Integer year1 = 0;
        Integer year2 = 0;
        Integer month1 = 99999;
        Integer month2 = 99999;
        Integer type1 = 99999;
        Integer type2 = 99999;

        if(rs.year != null && year != null) {

            if(!rs.year.equals(""))
                year1 = Integer.parseInt(rs.year);
            if(!year.equals(""))
                year2 = Integer.parseInt(year);
        }

        if(rs.month != null && month != null) {
            if(!rs.month.equals(""))
                month1 = Integer.parseInt(rs.month);
            if(!month.equals(""))
                month2 = Integer.parseInt(month);
        }

        if(rs.type == null)
            type1 = 99999;
        else
            type1 = 0;

        if(type == null)
            type2 = 99999;
        else
            type2 = 0;

        x = type2.compareTo(type1);
        if(x != 0) {
            return x;
        }

        x = year1.compareTo(year2);
        if(x != 0) {
            return x;
        }

        x = month1.compareTo(month2);
        if(x != 0) {
            return x;
        }


    return x;
}

3 个答案:

答案 0 :(得分:3)

我会重构(例如扔掉)您的完整代码并使用CompareToBuilder替换它。

这将创建以下代码:

enum ArticleType {
    ONLINE, PAPER
}

class Article implements Comparable<Article> {

    int year;
    int month;
    ArticleType type;

    Article(int year, int month, ArticleType type) {
        this.year = year;
        this.type = type;
        this.month = month;
    }

    @Override
    public int compareTo(Article o) {
        return new CompareToBuilder()
                .append(this.type, o.type)
                .append(this.year, o.year)
                .append(this.month, o.month)
                .toComparison();

    }

    @Override
    public String toString() {
        return Objects.toStringHelper(this)
                .add("month", month)
                .add("year", year)
                .add("type", type)
                .toString();
    }
}

@Test
public void testSortArticles() throws Exception {
    List<Article> articleList = new ArrayList<>();
    articleList.add(new Article(2012, 1, ArticleType.ONLINE));
    articleList.add(new Article(2011, 1, ArticleType.ONLINE));
    articleList.add(new Article(2011, 6, ArticleType.ONLINE));
    articleList.add(new Article(2010, 1, ArticleType.ONLINE));
    articleList.add(new Article(2010, 1, ArticleType.PAPER));
    articleList.add(new Article(2010, 2, ArticleType.PAPER));
    articleList.add(new Article(2010, 3, ArticleType.PAPER));
    articleList.add(new Article(2012, 1, ArticleType.PAPER));
    articleList.add(new Article(2012, 9, ArticleType.PAPER));

    Collections.sort(articleList);

    System.out.println(articleList);
}

打印此内容将导致以下结果:

[Article{month=1, year=2010, type=ONLINE}, Article{month=1, year=2010, type=PAPER},
 Article{month=2, year=2010, type=PAPER}, Article{month=3, year=2010, type=PAPER},
 Article{month=1, year=2011, type=ONLINE}, Article{month=6, year=2011, type=ONLINE},
 Article{month=1, year=2012, type=ONLINE}, Article{month=1, year=2012, type=PAPER},
 Article{month=9, year=2012, type=PAPER}]

将提供一个排序很好的列表。离线/在线也使用Enum(ArticleType)进行排序。在我看来,这看起来比你当前的实现要好一些。

答案 1 :(得分:1)

如果类型不为null,则将其替换为0,无论其值是什么。因此,将“纸张”与“在线”进行比较会导致将0与0进行比较,这显然是错误的。

我的第一个建议是使用适当的类型而不是String。月份和年份应为int s,类型应为枚举。你也应该努力使它们不可为空。

完成后,比较方法将减少到

public int compareTo(Article other) {
    int result = this.type.compareTo(other.type);
    if (result == 0) {
        result = Integer.compare(this.year, other.year);
    }
    if (result == 0) {
        result = Integer.compare(this.month, other.month);
    }
    return result;
}

请注意,使用枚举可以指定类型比较方式,只需按所需顺序列出值即可。

甚至更好,与番石榴:

public int compareTo(Article other) {
    return ComparisonChain.start()
                          .compare(this.type, other.type)
                          .compare(this.year, other.year)
                          .compare(this.month, other.month)
                          .result();
}

如果值可以为空,则必须将上述代码更改为

public int compareTo(Article other) {
    return ComparisonChain.start()
                .compare(this.type, other.type, Ordering.natural().nullsLast())
                .compare(this.year, other.year, Ordering.natural().nullsLast())
                .compare(this.month, other.month, Ordering.natural().nullsLast())
                .result();
}

答案 2 :(得分:0)

要回答您的问题,您只需使用compareTo()来比较两个String个对象。您只需确保按照所需的顺序对它们进行比较。做

x = type2.compareTo(type1);

降序比较String s。如果要按升序排序,则需要执行

x = type1.compareTo(type2);

此外,我对所有== null!= null项检查感到好奇。由于这是您自己的类,因此您应该控制成员字段是否为null。如果我正在编写类似的类,我会要求所有字段都由构造函数初始化,这样我就不需要检查null值。这可以大大简化类中的所有其他方法,包括此compareTo()方法。

此外,您应该更喜欢内置类型而不是包装类。换句话说,使用int s作为年份和月份字段而不是Integer s。这也可以通过几种方式帮助简化代码。首先,您不必担心null值。其次,您可以使用简单的减法与int进行比较:

int compareYears = this.year - rs.year;
if (compareYears != 0) {
    return compareYears;
}

根据这些建议,您不仅可以解决当前问题,还可以将代码行数减少至少一半。

(请注意,一般情况下,由于溢出,您必须小心使用减法来比较int。在这种情况下,由于我们比较年份,因此两个值都不应为负数,因此不应该有当然,你班上的其他人应该强制year字段实际上是有效年份。)