从ArrayList创建一个n-gram列表

Question

我试图从包含大量单词的ArrayList中创建一个N-gram列表。 我当前的代码如下所示：

public void ngram(int n){ 
    int i = 0; 
    for(String a : arraylist){ 
     if (i < arraylist.size() - 2) { 
    ngramlist.add(arraylist.get(i) + " " + arraylist.get(i+1) + " " + arraylist.get(i+2)); 
    i++; 
    } 
    } 
    System.out.println(ngramlist); 
} 

其中的ArrayList是用言语和ngramlist列表是列表我想从数组列表创建。 此代码有效，但不是我想要的。如果我输入像“猫狗鸟椅桌”这样的文字，输出将是“猫狗鸟，狗鸟椅，鸟椅桌”。 我试图完成的事情是，我可以使用变量n输入我想要在一行中添加的字数。所以如果我把2换成n，那它就是“猫狗，狗鸟，鸟椅，椅子桌子”。

所以我想我的代码是这样的：

public void ngram(int n){ 
    int i = 0; 
    for(String a : arraylist){ 
     if (i < arraylist.size() - n) { 

    (add all the words between i and i+n, so if n is 3 it adds 0-2, 1-3, 2-4 etc.) 

    i++; 
    } 
    } 
    System.out.println(ngramlist); 
} 

Answer 1

所以你可以使用里面另一个循环的主要原因之一：

String ngram = ""; 
for (int j = 0; j < n-1; j++) 
    ngram += arraylist.get(i + j) + " "; 
ngram += arraylist.get(i + n - 1); 
ngramlist.add(ngram); 

我同意这可能是低效的，因为它创建了很多字符串，但否则它很简单，它应该工作。 （编辑：实际上，我在想，如果它们存储在字符串池中，它可能不应该那么糟糕）

Answer 2

使用StringBuilder：

for(String a : arraylist){ 
    if (i < arraylist.size() - n) { 
     StringBuilder sb = new StringBuilder(); 
     for (int j = 0; j < n-1; j++) { 
      sb.append(arraylist.get(i+j)).append(" "); 
     } 
     sb.append(arraylist.get(i+n-1)); 
     ngramlist.add(sb.toString()); 
     i++; 
    } 
}