35. Search Insert Position

[dxxsxsxkx]

問題

https://leetcode.com/problems/search-insert-position/

次の問題

153. Find Minimum in Rotated Sorted Array

[potrue]

ここでif (begin == end) {return end + 1}としてしまってもいいと思いました。

[potrue]

良い代替案が思いつくわけではないのですが、answerという名前の変数が頻繁に変わるのは少し違和感があるかもしれません。

[nodchip]

end の位置には target より小さい値があるため、 begin でないとまずいように思いました。

[mamo3gr]

[nits]

Suggested change

	時間計算量 $O(N \log N)$ の制約があるので二分探索でやることが求められているんだろう。`step1.cpp` に書いた。
	時間計算量 $O(\log N)$ の制約があるので二分探索でやることが求められているんだろう。`step1.cpp` に書いた。

[mamo3gr]

end より右は常にtrue (nums[i] >= target) ということですね。

[mamo3gr]

begin より左側は常にfalseということですね。

[mamo3gr]

答えが [begin, end] の中にある状態を保つ

なのに

begin > end となったら終了

というのは違和感がありました。言葉通り捉えると、解が存在しないように思います。

[mamo3gr]

探索の3パターンを書き分けてみるとより理解が深まると思います。
ちなみに、私は ng (nums[i] < target) と ok (target <= nums[i]) の境界を求める、と考えるのが一番分かりやすかったです。

#include <vector>

class Solution {
public:
    int searchInsert(std::vector<int>& nums, int target) {
        int ng = -1;
        int ok = nums.size();

        while (ok - ng > 1) {
            int mid = ng + (ok - ng) / 2;
            if (nums[mid] >= target) {
                ok = mid;
            } else {
                ng = mid;
            }
        }

        return ok;
    }
};

参考：https://qiita.com/drken/items/97e37dd6143e33a64c8c