java - 字符串中按字典顺序排列的第 k 个最小字符

Question

我有一个用户输入的字符串和一个整数k。如何从输入字符串中获取按字典顺序排列的第 k 个最小字符？

这是我一直在尝试的

static Character find(int K, String s){
    SortedSet<Character> ss = new TreeSet<Character>();
    for(Character c : s.toCharArray()){
        ss.add(c);
    }
    Iterator it = ss.iterator();
    int i=0;
      while (it.hasNext()) {
         Object element = it.next();
         if(i==K-1)
             return (Character) element;

      }
      return null;
}

Answer 1

你的代码没问题，除了一件事:i的处理多变的。您应该在 while 的末尾增加它循环以使其匹配 K - 1 .

还有一种对 java8 更友好的方法来执行相同的操作:

String str = "hello";
int k = 2;

Optional<Character> kth = 
    Stream.of(str.split(""))
        .map(c -> c.charAt(0))
        .sorted()
        .distinct()
        .skip(k - 1)
        .findFirst();

System.out.println(kth.isPresent() ? kth.get() : "k too big!"); // h

此代码假设 k是从 1 开始的，并且您在获取第 k 个字符时不考虑重复项。

此外，正如评论中所指出的，您不应该为 Iterator 使用原始类型。考虑将其更改为 Iterator<Character> ，因此您不必强制转换 it.next() 返回的值.