AtCoder Beginner Contest 219C 間違っている箇所を知りたい

Question

AtCoder Beginner Contest 219C(こちら)がWAとなってしまいます。
どの部分が間違っているかご教授いただけないでしょうか。

import java.util.*;

public class Main {
  static List<Integer> l=new ArrayList<Integer>();
  public static void main(String[]args){
    Scanner sc=new Scanner(System.in);
    String x=sc.next();
    String[] xa=x.split("");
    Map<String,Integer> xmap=new HashMap<>();
    for(int i=0;i<26;i++){
      xmap.put(xa[i],i+1);
    }
    int n=sc.nextInt();
    List<List<String>> s=new ArrayList<List<String>>();
    Map<Long,String> ansmap=new HashMap<>();
    String ss[]=new String[n];
    for(int i=0;i<n;i++){
      ss[i]=sc.next();
      List sa=new ArrayList<>(Arrays.asList(ss[i].split("")));
      s.add(sa);
    }
    long ans[]=new long[n];
    for(int i=0;i<n;i++){
      long a=0;
      for(int j=0;j<s.get(i).size();j++){
        a+=xmap.get(s.get(i).get(j))*Math.pow(26,9-j);
      }
      ans[i]=a;
      ansmap.put(ans[i],ss[i]);
    }
    Arrays.sort(ans);
    for(int i=0;i<n;i++){
      System.out.println(ansmap.get(ans[i]));
    }
  }
}

Answer 1

並べ替えですので Comparator だけ作ればよいのでは。

import java.util.*;

public class Main {
  public static void main(String[] args) {
    execute("bacdefghijklmnopqrstuvwxzy", Arrays.asList("abx", "bzz", "bzy", "caa"));
    System.out.println("----");
    execute("zyxwvutsrqponmlkjihgfedcba", Arrays.asList("a","ab","abc","ac","b"));
    System.out.println("----");
    execute("abcdefghijklmnopqrstuvwxyz", Arrays.asList("az","b"));
  }
  static void execute(String dic, List<String> names) {
    Collections.sort(names, (a, b) -> {
      int length = Math.min(a.length(), b.length());
      for(int i=0; i<length; i++) {
        int v = dic.indexOf(a.charAt(i)) - dic.indexOf(b.charAt(i));
        if(v != 0) return v;
      }
      return a.length() - b.length();
    });
    for(String name : names) System.out.println(name);
  }
}

bzz
bzy
abx
caa
----
b
a
ac
ab
abc
----
az
b

Comments

[den7]

返信が遅くなってしまい申し訳ございません、。
こちらの方がスマートですね！
大変勉強になりました。ありがとうございます！

Answer 2

こちらの入力例でデバッグしてみてください。

abcdefghijklmnopqrstuvwxyz
2
az
b

Comments

[den7]

ありがとうございます!ACできました。累乗のところがよくなかったみたいです。