אולי תוכלו לפרסם את הפתרון כתוב בקוד?

פעם לפני שנים בראיון שאלו אותי אותה השאלה עם סיבוך נוסף: מותר לראות את הצבע של כל כדור פעם אחת בלבד, כלומר ברגע שבדקנו כדור פעם אחת לא נוכל לבדוק אותו שוב

באיזה חברה שאלו את זה?

השיטה הראשונה יותר יעילה בגלל שאפשר לדלג על הכדורים האדומים בריצה השניה.

שלב ראשון – נעביר את כל הכדורים האדומים לתחילת המערך: א. נחזיק אינדקס A שמצביע על תחילת המערך ואינדקס B שמצביע על סוף המערך. ב. נקדם את אינדקס A לכיוון סוף המערך, עד שניתקל בכדור שאינו אדום. ג. נקדם את אינדקס B לכיוון תחילת המערך, עד שניתקל בכדור אדום. ד. נחליף בין הכדורים שמוצבעים ע"י האינדקסים. ה. נחזור על שלבים ב', ג', ו-ד', עד שאינדקס A יעבור את אינדקס B. שלב שני – באופן דומה, נעביר את כל הכדורים הירוקים לסוף המערך... סיבוכיות זמן – O(n), סיבוכיות זיכרון – O(1).