Q 学習は、各状態で各行動がどれだけ良いか Q(s, a) を経験だけから学習します。状態 s で行動 a を取り、reward を受け取って状態 s' に移ったら:
- エピソードが終了した場合 (
done= 1) は target = reward - そうでなければ target = reward + gamma × (a' についての Q(s', a') の最大値)
- Q(s, a) ← Q(s, a) + alpha × (target - Q(s, a))
すべての Q 値を 0 から始め、遷移を順に適用します。
q_learning(num_states, num_actions, transitions, alpha, gamma) を実装してください。
入力
num_states, num_actions: 表の大きさ。
transitions: 1 ステップにつき 1 行の [state, action, reward, next_state, done]。
alpha: 学習率。
gamma: 割引率。
出力
状態ごとに 1 行、行動ごとに 1 列の行列として Q 表を返します。
制約
1 ≤ num_states, num_actions ≤ 100、0 ≤ 遷移の数 ≤ 10,000
-100 ≤ reward ≤ 100、done は 0 または 1
0 < alpha ≤ 1、0 ≤ gamma ≤ 1
例
入力:
num_states = 2
num_actions = 2
transitions = [
[0, 1, 0, 1, 0],
[1, 0, 10, 0, 1]
]
alpha = 0.5
gamma = 0.9
期待される出力:
[
[0.0, 0.0],
[5.0, 0.0]
]
解説
最初のステップではまだ報酬がありません。target = 0 + 0.9 × max(Q[1]) = 0 なので Q[0][1] は 0 のままです。2 番目のステップは報酬 10 でエピソードを終えます。Q[1][0] = 0 + 0.5 × (10 - 0) = 5。
入力:
num_states = 2
num_actions = 2
transitions = [
[1, 0, 10, 0, 1],
[0, 1, 0, 1, 0]
]
alpha = 0.5
gamma = 0.9
期待される出力:
[
[0.0, 2.25],
[5.0, 0.0]
]
解説
同じステップを逆の順に適用します。先に Q[1][0] が 5 になるので、状態 0 からのステップはそれを参照できます。target = 0 + 0.9 × 5 = 4.5、Q[0][1] = 0.5 × 4.5 = 2.25。報酬が 1 つ前の状態へ伝わりました。
参考資料
-
例 1 未実行入力:
num_states = 2 num_actions = 2 transitions = [ [0, 1, 0, 1, 0], [1, 0, 10, 0, 1] ] alpha = 0.5 gamma = 0.9期待される出力:[ [0.0, 0.0], [5.0, 0.0] ]あなたの出力 -
例 2 未実行入力:
num_states = 2 num_actions = 2 transitions = [ [1, 0, 10, 0, 1], [0, 1, 0, 1, 0] ] alpha = 0.5 gamma = 0.9期待される出力:[ [0.0, 2.25], [5.0, 0.0] ]あなたの出力
提出時に非公開テスト4件も実行されます。 失敗したテストがあれば、そのテストが表示されます。チャレンジを解くとすべて見られます。
Online Java Compiler
Write, compile, run, and debug Java code online in a full-featured browser IDE. Work with multiple files, save your workspace, and share or collaborate in real time - no setup required.
☕ このJavaオンライン実行ツールについて
CodeUtility Java Executorは、ブラウザ上で直接Javaプログラムを記述・実行できるツールです - インストールやJDKの設定、IDEは不要です。安全なサンドボックス上で動作し、実際のJavaランタイム バージョン8、11、17、21、そして最新に対応しています。
このツールは、本物のjavacコンパイラとJavaランタイム環境をクラウドで使用してコードをコンパイル・実行します。
そのため、クラスやメソッド、ロジックをローカルとまったく同じ挙動でテストできます。
オブジェクト指向の練習、面接対策、統合前のスニペット検証など、あらゆる用途で CodeUtility Java Executorは高速かつ信頼できる実行環境を提供し、どんなデバイスからでもすぐに本物のJavaコードを実行できます。
同じ実行環境で、複数ファイル対応エディター、分離された実行ターミナル、ランタイム別ライブラリ、永続ワークスペース、共有、リアルタイム共同作業を利用できます。
この実行環境の使い方
ランタイムを選択し、ファイルを開くか作成してコードを記述し、[実行]を選択します。出力はターミナルに表示されます。サインインしたユーザーは、ワークスペースにファイルを保存し、ライブラリをインストールして、共有を通じて共同作業できます。
- バージョンを選択して現在のファイルを編集するか、ワークスペース内の別のファイルを開きます。
- コードを実行し、求められた場合はターミナルに入力して、保存された実行履歴を確認します。
- 複数のコードスニペットが必要な作業では、ライブラリ、ワークスペースツール、共有、チャットを利用します。
💡 上のエディタで試せるJavaの基本とサンプル
1. 変数と定数の宣言
Javaでは各変数の型を宣言する必要があります。定数にはfinalを使います。
int age = 30;
double pi = 3.14159;
char grade = 'A';
String name = "Alice";
boolean isActive = true;
// 定数
final int MAX_USERS = 100;
final String COMPANY = "CodeUtility";
2. 条件分岐(if / switch)
制御フローにはif、else if、switchを使います。
int x = 2;
if (x == 1) {
System.out.println("1");
} else if (x == 2) {
System.out.println("2");
} else {
System.out.println("その他");
}
switch (x) {
case 1:
System.out.println("1");
break;
case 2:
System.out.println("2");
break;
default:
System.out.println("その他");
}
3. ループ
繰り返しにはfor、while、do-whileを使います。
for (int i = 0; i < 3; i++) {
System.out.println(i);
}
int n = 3;
while (n > 0) {
System.out.println(n);
n--;
}
4. 配列
配列は同じ型の要素を固定長で保持します。
int[] numbers = {10, 20, 30};
System.out.println(numbers[1]);
5. ArrayListの操作
ArrayListは可変長のリストに使います。
import java.util.ArrayList;
ArrayList<Integer> nums = new ArrayList<>();
nums.add(1);
nums.add(2);
nums.add(3);
nums.remove(Integer.valueOf(2));
for (int num : nums) {
System.out.print(num + " ");
}
6. コンソール入出力
入力にはScanner、出力にはSystem.outを使います。
import java.util.Scanner;
Scanner scanner = new Scanner(System.in);
System.out.print("お名前を入力してください: ");
String name = scanner.nextLine();
System.out.println("こんにちは、" + name);
7. メソッド
戻り値の型と引数を指定してメソッドを定義します。
public static int add(int a, int b) {
return a + b;
}
System.out.println(add(3, 4));
8. HashMap
HashMapはキーと値のペアを格納するために使います。
import java.util.HashMap;
HashMap<String, Integer> ages = new HashMap<>();
ages.put("Alice", 30);
System.out.println(ages.get("Alice"));
9. 例外処理
実行時例外はtryとcatchで処理します。
try {
throw new Exception("問題が発生しました");
} catch (Exception e) {
System.out.println(e.getMessage());
}
10. ファイル入出力
ファイル操作にはjava.nio.fileのFilesとPathsを使います。
import java.nio.file.*;
import java.io.IOException;
Files.writeString(Paths.get("file.txt"), "こんにちは、ファイル");
String content = Files.readString(Paths.get("file.txt"));
System.out.println(content);
11. 文字列操作
Javaの文字列はlength()、substring()、contains()など多くのメソッドをサポートします。
String text = "Hello World";
System.out.println(text.length());
System.out.println(text.substring(0, 5));
System.out.println(text.contains("World"));
12. クラスとオブジェクト
Javaはクラスとインスタンスを用いたオブジェクト指向プログラミングをサポートします。
class Person {
String name;
Person(String name) {
this.name = name;
}
void greet() {
System.out.println("こんにちは、私は " + name);
}
}
Person p = new Person("Alice");
p.greet();