Java:在 3 行或更少的时间内查找 2 个集合的并集、交集、差异和异或

Question

对于一个项目，我需要找到两个集合的并集、交集、差异和异或。每个方法不能超过3行代码。到目前为止，我已经有了工作方法，但我一生都看不到如何将差异和异或方法压缩到 3 行或更少的代码。这是我到目前为止所拥有的:

static TreeSet<String> union( TreeSet<String> set1, TreeSet<String> set2 )
{
    TreeSet<String> s1 = new TreeSet<String>(set1);
    s1.addAll(set2);
    return s1;
}

static TreeSet<String> inter( TreeSet<String> set1, TreeSet<String> set2 )
{
    TreeSet<String> s1 = new TreeSet<String>(set1);
     s1.retainAll(set2);
    return s1; 
}

static TreeSet<String> diff( TreeSet<String> set1, TreeSet<String> set2 )
{

    TreeSet<String> s1= new TreeSet<String>(inter(set1, set2));
    TreeSet<String> s2 = new TreeSet<String>(set1);
    s2.removeAll(s1);
    return s2; 
}

static TreeSet<String> xor( TreeSet<String> set1, TreeSet<String> set2 )
{
    TreeSet<String> s1 = new TreeSet<String>(union(set1,set2));
    TreeSet<String> s2 = new TreeSet<String>(inter(set1, set2));
    s1.removeAll(s2);
    return s1; 
}

我一直盯着这个，但就是没有想到。在此先感谢您的时间。我知道这可能是一个简单的解决方案，但我只是没有看到。

Answer 1

如果您不使用辅助变量之一，则可以每行保存一行。相反，写

s2.removeAll(new TreeSet<String>(inter(set1, set2)));

您的 diff 方法，以及

s1.removeAll(new TreeSet<String>(inter(set1, set2)));

用于您的异或方法。

编辑:

s2.removeAll(inter(set1, set2));

和

s1.removeAll(inter(set1, set2)));

也可以工作，因为你的 inter() 返回一个 TreeSet。