区間ごとの合計を求めたい

Question

String[][] a = {{"a", "1"}, {"a", "2"}, {"a", "3"}, {"b", "11"}, {"b", "12"}, {"b", "13"}};のような多次元配列が与えられているときに
a[i][0]が同じ場合、その区間の合計を求めたいです。
理想とする結果は、sum = {1, 3, 6, 11, 23, 35}のような結果を得たいです。
よろしくお願いします。

public class sample3 {
	public static void main(String[] args) {
		
		String[][] a = {{"a", "1"}, {"a", "2"}, {"a", "3"}, {"b", "11"}, {"b", "12"}, {"b", "13"}};
		int[] sum = new int[a.length];
		int i = 0;
		sum[0]=Integer.parseInt(a[i][1]);
		
		for(i=0; i < a.length-1; i++) {
			if(a[i][0] == a[i+1][0]) {
				sum[i+1] = sum[i]+Integer.parseInt(a[i][1]) + Integer.parseInt(a[i+1][1]);
				System.out.println(sum[i]);
			}
		}
	}
		
}

Comments

[jimbe]

> a[i][0]が同じ場合、その区間の合計を求めたい
> sum = {1, 3, 6, 11, 23, 35}
"a" の合計, "b" の合計 でしたら
sum = { 6, 36 }
ではないのでしょうか?

[swordone]

キーごとに累積和みたいなことをしたいってこと？

[LouiS0616]

(a 3) (b 1) (a 4) のとき、次のどちらが望ましいのでしょう？
・3 1 4
・3 1 7

[negoto]

わかりづらくてすみません、jimbeさん、swordoneさんの理解が正しいです。
同キーごとの和を求めたいです。

[swordone]

LouiS0616さんの質問にも答えて欲しいですね。つまり、キーが連続でない場合、かまわず和を取るのか、仕切り直すのか？

[negoto]

すみません、質問の意味がわからなかったので、答えられませんでした・・・
キーの並びがa, a, b, aの場合、後のaの値も合計を取るのか、という認識で話を進めますが、キーはソートしているので後で既出のキーは出てきません。

[LouiS0616]

> キーはソートしているので後で既出のキーは出てきません。

物凄く重要な情報です。
そのような暗黙の前提が閲覧者に伝わるよう配慮できるとより良い質問をすることができます。

[negoto]

すみません、具体例を出しているから伝わるだろうと思いました。
以後気を付けます。
加えて、その条件でコードがどうなるか教えてください。

Answer 1

計算ロジックと配列の扱いがわかりにくい場合は、一時変数の中にいれて扱うとデバッグしやすいでしょう。

	public static void main(String[] args) {

		String[][] a = { { "a", "1" }, { "a", "2" }, { "a", "3" }, { "b", "11" }, { "b", "12" }, { "b", "13" } };
		int[] sum = new int[a.length];
		int i = 0;
		int temporary = Integer.parseInt(a[0][1]);
		sum[0] = temporary;
		for (i = 0; i < a.length - 1; i++) {
			if (a[i][0] == a[i + 1][0]) {
				temporary = temporary + Integer.parseInt(a[i + 1][1]);
			} else {
				temporary = Integer.parseInt(a[i + 1][1]);
			}
			sum[i + 1] = temporary;
		}

		for (i = 0; i < sum.length; i++) {
			System.out.println(sum[i]);
		}
	}

Comments

[negoto]

一度に全てをやろうとしていました！そうではなく、一時変数にi、i+1番目の合計を代入してから、配列sum[i+1]に合計を代入しているのですね！
ありがとうございます！