共同好友（java）

查询两个用户的共同好友（例如两个QQ用户）
例如有两个用户，他们分别有一个好友集，user1friends和user2friends，分别放到set集合中

import java.util.Set;
import java.util.TreeSet;

public class SameFriends {
    public static Set<Integer> intersection(Set<Integer> user1friends,Set<Integer> user2friends){
    //判断用户的好友是否为空
        if((user1friends == null) || user1friends.isEmpty()){
            return null;
        }
        if((user2friends == null) || user2friends.isEmpty()){
            return null;
        }
        //判断好友数量，以用来做性能优化
        if(user1friends.size() < user2friends.size()){
            return intersect(user1friends, user2friends);
        }else{
            return intersect(user2friends, user1friends);
        }
    }

    public static Set<Integer> intersect(Set<Integer> smallset,Set<Integer> largeset){
        Set<Integer> result = new TreeSet<Integer>();
        //迭代处理小集合来提高性能
        for (Integer x : smallset) {
            if(largeset.contains(x)){
                result.add(x);
            }
        }
        return result;
    }
}

测试

import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;

public class Test1 {
    public static void main(String[] args) {
        Set user1friends = new TreeSet();
        Set user2friends = new TreeSet();
        //用户1创建9个好友，为10,20,30......
        for (int i = 1; i < 10; i++) {
            user1friends.add(i*10);
        }
        //用户2创建9个好友，为5,10,15......
        for (int i = 1; i < 10; i++) {
            user2friends.add(i*5);
        }
        Set result = SameFriends.intersection(user1friends, user2friends);
        Iterator it = result.iterator();
        while (it.hasNext()) {
            System.out.println((it.next()));

        }
    }
}

输出结果为：

10
20
30
40