Javaメソッド

Question

引数で指定した数だけ草を生やして文字列として返す関数 kusa を作成する
以下のコードをもっと素直に書く方法としてはどんなものがあるのか知りたいです
よろしくお願いします

public class W_Unavoidable {
    public static void main(String[] args) {
        String w = kusa(50);
        System.out.println(w);
    }
    public static String kusa(int num){
        String grass = "";
        for(; num > 0; num--){
            grass = grass + "w";
        }
        return grass;
    }

Answer 1

public class Main {
    public static void main(String[] args) throws Exception {
        System.out.println(kusa(50));
    }
    
    public static String kusa(int n){
        if (n < 1) return "";
        return "w" + kusa(--n);
    }
}

計算量やメモリやスタックのことなんかは知りません。
コードが美しければよいのですｗ

Answer 2

標準ライブラリだけでの実装を6個ほど考えました。

/** StringBuilder版
 * 高速なStringBuliderに対して文字を追加する。
 * 初期のキャパシティをnumにしておくと良いような気がする。
 */
public static String kusa1(int num){
    StringBuilder grass = new StringBuilder(num);
    for (int i = 0; i < num; i++) {
        grass.append("w");
    }
    return grass.toString();
}

/** StringBuilder + 2倍しながら追加版
 * 高速なStringBuliderに追加する。
 * 2倍しながら追加していくので、O(n)=>O(log n)になるはず。
 */
public static String kusa2(int num){
    StringBuilder grass = new StringBuilder(num);
    String w = "w";
    while (num > 0) {
        if ((num & 1) == 1) {
            grass.append(w);
        }
        w += w;
        num >>= 1;
    }
    return grass.toString();
}

/** char[]版
 * charの配列で作ってからStringにする。
 */
public static String kusa3(int num){
    char[] grass = new char[num];
    for (int i = 0; i < num; i++) {
        grass[i] = 'w';
    }
    return new String(grass);
}

/** char[] + Arrays.fill版
* charの配列で作ってからStringにする。
 * Arrays.fillで埋める。
 */
public static String kusa4(int num){
    char[] grass = new char[num];
    java.util.Arrays.fill(grass, 'w');
    return new String(grass);
}

/** IntStream版
 * なんとなくIntSteramを使いたかっただけ。
 */
public static String kusa5(int num){
    StringBuilder grass = new StringBuilder(num);
    java.util.stream.IntStream.range(0, num).forEach((i) -> {
        grass.append("w");
    });
    return grass.toString();
}

/** IntStream + collect版
 * なんとなくIntSteramを使いたかっただけ、その2。
 * collectを使ってみたけど、面倒。parallelにすると壊れそう。
 */
public static String kusa6(int num){
    return java.util.stream.IntStream.range(0, num).collect(
        () -> new StringBuilder(),
        (grass, i) -> grass.append("w"),
        (a, b) -> a.append(b.toString())
    ).toString();
}

Stringに+するよりStringBuiderにappend()した方が高速です。kusa1にするだけでもかなり違うと思います。StringBuilder.appendが定量計算時間なら、kusa2が一番速くなりそうですが、kusa4あたりが最速のような気がします。二文字以上だとkusa3とkusa4は使えませんが、一番素直なやり方かかも知れません。kusa5とkusa6は、まぁ、IntStreamを使いたかっただけかと。

Comments

[swordone]

Stream.generateからlimitは…？

[raccy]

うーん、それならIntStreamを'w'だけのストリームにしてappendCodePointかなー

[退会済みユーザー]

ありがとうございます！参考にさせていただきます！

[swordone]

連結する文字数を10000000にして、実際にかかる時間を計測してみました。
kusa3とkusa4は15ms前後、他は3桁msかかります。
速くなりそうというkusa2は、連結する文字列を倍々にするための連結にString生成をしているため、余計な時間がかかるのだと思われます。
kusa1はappendの引数を"w"ではなく'w'にすると、70ms前後になりました。

Answer 3

public class W_Unavoidable {
	   public static void main(String[] args) {
	        kusa(50);
	    }
	    public static void kusa(int num){
	        for(int i = 0;i<num ; i++){
	        	System.out.print("w");
	        }
	    }
}

あえてforを使っていくスタイル マイナスが素直じゃない感じなので。

Comments

[raccy]

kusaは「文字列として返す関数」なので、出力してしまってはだめなのでは？

[toutou]

あ、ほんとだ。printは確認用か。条件だけ変更ってことで。

[退会済みユーザー]

ありがとうございます！参考になりました

Answer 4

StringUtils.repeatメソッドでできそうですね。

Comments

[退会済みユーザー]

ありがとうございます！参考にします

Answer 5

https://commons.apache.org/proper/commons-lang/apidocs/org/apache/commons/lang3/StringUtils.html
のStringUtilsを使ってよいのならすごいスッキリかけます。

public static String kusa(int num){
    return StringUtils.repeat("w", num);
}