Question

import java.util.Arrays;
import java.util.Scanner;

public class SuffixArray {
public static class Tuple implements Comparable<Tuple>{
    public Integer originalIndex;
    public Integer firstHalf;
    public Integer secondHalf;
@Override
    public int compareTo(Tuple o) {
        if(this.firstHalf.compareTo(o.firstHalf)==0){
            return this.secondHalf.compareTo(o.secondHalf);
        }else {
            return this.firstHalf.compareTo(o.firstHalf);
        }
    }
}
public static void main(String args[]){
    Scanner scanner = new Scanner(System.in);
    String input = scanner.next();
    int n = input.length();
    Tuple[] tuples = new Tuple[n];
    int[][] suffixRank = new int[100][100];
    for (int i = 0; i < n; i++){
        tuples[i] = new Tuple();
        suffixRank[0][i] = input.charAt(i) - 'a';
    }
    for(int step=1, count=1;count < n;count*=2, step++){
        for (int i = 0; i<n;i++){
            tuples[i].firstHalf = suffixRank[step-1][i];
            tuples[i].secondHalf = i + count < n ? suffixRank[step-1][i+count] : -1;
            tuples[i].originalIndex = i;
        }
        Arrays.sort(tuples);
        for (int i=0;i<n;i++){
            suffixRank[step][tuples[i].originalIndex] = i > 0 && tuples[i].firstHalf == tuples[i-1].firstHalf && tuples[i].secondHalf==tuples[i-1].secondHalf ? suffixRank[step][tuples[i].originalIndex] : i;
        }
    }
    for (int i=0;i<n;i++){
        System.out.print(tuples[i].originalIndex + " ");
    }
  }
}

上述程序是使用nlogn算法的后缀数组。我已经从资源中应用了相同的算法，该资源是下面发布的链接。如果我在C ++中应用相同的算法，它工作正常但是当我在Java中应用相同的代码时，它给了我错误的输出。它给了我错误的元组索引。我无法弄清楚这段代码有什么问题，所以请帮助我。

Suffix Array

测试用例1：

输入：abaab 输出：3 2 0 4 1 预期产出：2 3 0 4 1

测试用例2：

输入：香蕉输出：4 5 3 1 0 2 预期产出：5 3 1 0 4 2

Answer 1

不知道这是否是代码的唯一问题，但在排序后的for循环中会出现一个小的不准确性。具体来说，在循环中的if子句之后，你应该交换

? suffixRank[step][tuples[i].originalIndex] : i;

用这个：

? suffixRank[step][tuples[i-1].originalIndex] : i;

我创建了一个后缀数组，但我不知道这段代码有什么问题

1 个答案: