入力された整数a,bを含めたその間の全整数の和を求めて返すプログラム

Question

###前提・実現したいこと
入力された整数a,bを含めたその間の全整数の和を求めて返すプログラムを書いています。

a = 3, b = 5の時、sum = 12
a = 6, b = 4の時、sum = 15

###発生している問題・エラーメッセージ
エラーなどは起こっていないのですが、これが最適解かどうか意見をお聞きしたいです。もしより良いアルゴリズムがあれば教えていただけますと幸いです。

###該当のソースコード

package javaalgorithm;
import java.util.Scanner;

public class ex1_1_9 {

	public static void main(String[] args) {
		Scanner stdIn = new Scanner (System.in);
		System.out.println("aからbまでの和を求める。");
		System.out.print("aの値：");
		int a = stdIn.nextInt();
		System.out.print("bの値：");
		int b = stdIn.nextInt();

		int sum = 0;
		int mid = 0;
		int midsum =0;

		if(a<b){
		   for(int i =0; i<=(b-a-1); i++){
				mid = a++; //小さい方の値に1を足していく
				midsum += mid; //aとbの間の数の和を求める
			}
		}else if(a>b){
			for(int i =0; i<=(a-b-1); i++){
				mid = b++;
				midsum += mid;
			}
		}
		sum = a + midsum + b;
		System.out.println("aとb間の数の総和は" + sum + "です。");
	}

	}

###補足情報(言語/FW/ツール等のバージョンなど)
Eclipse Java EE IDE for Web Developers.
Version: Neon.3 Release (4.6.3)

Comments

[episteme]

最適解てなんですか? いちばん速い? いちばん短い? いちばんわかりやすい? ほかのなにか?

[unz.hori]

そうですね、epistemeさんも書かれてますがどういう観点での最適解を求めているのかによって回答が変わると思います。

Answer 1

等差数列の和には公式があります。
https://www.google.co.jp/search?ei=uK1mWraxHsyq0ASj2pSACA&q=等差数列の和&oq=等差数列の和
この公式を使っていただければ、O(1)ですから最速と言えます。

O(1)というのは入力した数字の大きさにかかわらず一定時間で解けるという意味です。質問文のプログラムは(b-a)の大きさに比例した時間がかかるのでO(n)の時間がかかると表現します。

Comments

[退会済みユーザー]

ご回答いただきましてありがとうございました。

Answer 2

• a > bとb < aのコードは同じなので共通化する → 必ずa < bにする。
• midやmidsumは不要 → ループカウンタをそのまま足すようにする。

// a > b なら入れ替える
if (a > b) {
    int tmp = a;
    a = b;
    b = tmp;
}

//aからbまで足していく
int sum = 0;
for (int n = a; n <= b; n++) {
    sum += n;
}

実際には公式を使うことをお勧めしますｗ

Comments

[退会済みユーザー]

ご回答いただきましてありがとうございました。

Answer 3

a == bのときsumがa + bになるのが気になります。

ちなみにstreamを使うと合計を出す部分は1行で書けます。

Scanner stdIn = new Scanner (System.in);
System.out.println("aからbまでの和を求める。");
System.out.print("aの値：");
int a = stdIn.nextInt();
System.out.print("bの値：");
int b = stdIn.nextInt();

// a == bの場合は考慮していない
if(a > b){
  int temp = a;
  a = b;
  b = temp
}

int sum = IntStream.rangeClosed(a,b).sum();

System.out.println("aとb間の数の総和は" + sum + "です。");

Comments

[退会済みユーザー]

ご回答いただきましてありがとうございました。