如何使用CompareTo()和TreeSet()

时间:2017-08-26 13:48:28

标签: java sorting duplicates compareto treeset

我有两个文件,其中包含以下行:ProductName(String); ProductCode(String); SellData(Int)。 两个文件中出现了一些ProductCode。我需要摆脱重复项并按ProductCode对项目进行排序。最后,输出应该是按代码排序的名称。我想使用CompareTo()和TreeSet()来解决这个问题,但我的程序有问题。这是:

class Gyumi implements Comparable<Gyumi> {
   String termek = "";
   String termekkod = "";
   int[] evek = new int[3];

  @Override
  public String toString() {
      return termek + ", " + termekkod + ", " + Arrays.toString(evek);
  }

  @Override
  public int compareTo(Gyumi t) {
      if(termekkod >((Gyumi)t.termekkod){
          return 1;
      }
      if(termekkod <((Gyumi)t.termekkod){
        return -1;
      }
        return 0;
    }
 }
}

public class Gyumolcs {

static ArrayList<Gyumi> lista = new ArrayList<Gyumi>();
static ArrayList<Gyumi> lista2 = new ArrayList<Gyumi>();
static TreeSet<Gyumi> halmaz = new TreeSet<Gyumi>();

public static void Feltolt() {
    File r = new File("termekkod_uzlet2.csv");
    File f = new File("termekkod_uzlet1.csv");
    try {
        Scanner scan = new Scanner(f, "iso-8859-2");
        while (scan.hasNextLine()) {
            String sor = scan.nextLine();
            String[] tomb = sor.split(";");
            Gyumi gy = new Gyumi();
            gy.termek = tomb[0];
            gy.termekkod = tomb[1];
            gy.evek[0] = Integer.parseInt(tomb[2]);
            gy.evek[1] = Integer.parseInt(tomb[3]);
            gy.evek[2] = Integer.parseInt(tomb[4]);
            lista.add(gy);
            halmaz.add(gy);
        }
    } catch (Exception e) {
        System.out.println("Hiba " + e.getMessage());
    }
    try {
        Scanner scan2 = new Scanner(r, "iso-8859-2");
        while (scan2.hasNextLine()) {
            String sor = scan2.nextLine();
            String[] tomb = sor.split(";");
            Gyumi gy = new Gyumi();
            gy.termek = tomb[0];
            gy.termekkod = tomb[1];
            gy.evek[0] = Integer.parseInt(tomb[2]);
            gy.evek[1] = Integer.parseInt(tomb[3]);
            gy.evek[2] = Integer.parseInt(tomb[4]);
            lista2.add(gy);
            halmaz.add(gy);
        }
    } catch (Exception e) {
        System.out.println("valami" + e.getMessage());
    }
}

public static void main(String[] args) {
    Feltolt();
    System.out.println(halmaz);
  }
}

1 个答案:

答案 0 :(得分:0)

您的 compareTo 方法错误,请将其更改为:

  @Override
  public int compareTo(Gyumi t) {
     if(termekkod.compareTo(t.termekkod)>0){
         return 1;
     }
     if(termekkod.compareTo(t.termekkod)<0){
         return -1;
     }
         return 0;
   }

另外,请在代码中遵循一般的OOPS设计。为您的成员变量提供私有范围,并使用getter / setter检索/设置它们。

OO设计看起来更整洁精确:

@Override
public int compareTo(Gyumi t) {
   if(termekkod.compareTo(t.gettermekkod())>0){
       return 1;
   }
   if(termekkod.compareTo(t.gettermekkod())<0){
       return -1;
   }
       return 0;
 }