Rust で usize 型を u32 型にする

Question

質問内容

Rust でベクトルを走査して、条件を満たすインデックスを得るということをしたいと思っています。

fn my_func(nums: Vec<i32>, target: i32) -> Vec<i32> {
  for i in 0..(nums.len() - 1) {
    for j in i..nums.len() {
      if nums[i] + nums[j] == target {
        return vec![i, j];
      }
    }
  }
}

しかし、i, jはusize型なので返り値の型がエラーになります。これはネット上の問題として解いているので返り値の型は出題側に決められていて変えることができません。なので何とかしてusize -> i32 にキャストしたいのですが方法がわかりません。

Rust 自体が初心者なのでこの考え方自体があっているかどうかも知りたいです。よろしくお願いします。

(追記) エラーメッセージ

error[E0308]: mismatched types
 --> src/main.rs:5:29
  |
5 |                 return vec![i, j];
  |                             ^ expected i32, found usize

error[E0308]: mismatched types
 --> src/main.rs:2:5
  |
1 |   fn my_func(nums: Vec<i32>, target: i32) -> Vec<i32> {
  |                                              -------- expected `std::vec::Vec<i32>` because of return type
2 | /     for i in 0..(nums.len() - 1) {
3 | |         for j in i..nums.len() {
4 | |             if nums[i] + nums[j] == target {
5 | |                 return vec![i, j];
... |
8 | |
9 | |     }
  | |_____^ expected struct `std::vec::Vec`, found ()
  |
  = note: expected type `std::vec::Vec<i32>`
             found type `()`

error: aborting due to 2 previous errors

For more information about this error, try `rustc --explain E0308`.

試したこと： as で usize -> i32 へ変換する

let ans1 = i as i32;
let ans2 = j as i32;

return vec![ans1, ans2];

をif文の中に入れたところ次のようなエラーが出ました。

error[E0308]: mismatched types
  --> src/main.rs:2:5
   |
1  |   fn two_sum(nums: Vec<i32>, target: i32) -> Vec<i32> {
   |                                              -------- expected `std::vec::Vec<i32>` because of return type
2  | /     for i in 0..(nums.len() - 1) {
3  | |         for j in i..nums.len() {
4  | |             if nums[i] + nums[j] == target {
5  | |                 let ans1 = i as i32;
...  |
10 | |
11 | |     }
   | |_____^ expected struct `std::vec::Vec`, found ()
   |
   = note: expected type `std::vec::Vec<i32>`
              found type `()`

error: aborting due to previous error                                                                                                                        
For more information about this error, try `rustc --explain E0308`.

Comments

[meg_]

質問のコードを実行すると、どのようなエラーが出るのでしょうか？

Answer 1

エラーの原因は2つです。

---

1. returnで返すvec!の型が宣言と異なる

これは質問者様もすでにご存知のようですので、型の変換についてだけ書きます。

Rustではasを使うことで特定の型同士の変換を行うことができます。
(asで何を変換できるかの詳細はReferenceに書かれています)

asを使って5行目の型を合わせると以下のようになります。

return vec![i as i32, j as i32];

今回は要素が2個しかないので個々にasを付けましたが、もし個数が多いようならイテレータを用いて以下のようにもできます。

return vec![i, j].into_iter().map(|n| n as i32).collect();

---

1. for文を抜けた後に返り値がない

引数の特性によりfor文の中で必ず関数が返るなら、for文の後に戻り値を書く必要はないのですがコンパイラにはそれはわかりません。
もしfor文を通り過ぎることがないと確信できるならば、8行目にunreachable!()を書き足すことで、コンパイラにそこに到達することはないと伝えることができます。
(unreachable!()に到達してしまった場合はpanicします)

---

以上の点を解消したコードは以下のようになります。

fn my_func(nums: Vec<i32>, target: i32) -> Vec<i32> {
    for i in 0..(nums.len() - 1) {
        for j in i..nums.len() {
            if nums[i] + nums[j] == target {
                return vec![i as i32, j as i32];
            }
        }
    }
    unreachable!()
}

Answer 2

戻り値を返さないコードパスがあるのが問題です。二重for文の中のif文の条件を一度も満たさなかった場合を考えてください。

Answer 3

Rust Documentation 型間のキャスト

let x: i32 = 5;
let y = x as i64;

上記例が載っていました。

Comments

[Yhaya]

as で変換することはすでに試しましたがエラーが出ます

[meg_]

既に試したこととその結果については質問欄へ追記いただくとより良い回答が得られるかと思います。

[Yhaya]

そうですね、ありがとうございます。