Нужно рассматривать самый неудачный вариант.
а) Если мы ждем 2 желтых шарика, то в самом худшем случае вытащим сначала 3 красных, а потом уже останутся только желтые.
Значит, для гарантии нужно вытащить 3+2=5 шариков
б) Нам нужны разные цвета, значит в худшем случае вначале вытащим 3 одноцветных, а 4-й шарик даст разноцветность с 2 предыдущими.
Значит, нужно вытащить 4 шарика.