Предлог на используется в сочетании со словом тема: вариация на тему. Ваш пример не кажется удачным.
Повторить предлог нужно.
Форма на корню имеет более узкое значение: на корню, в зн. нареч. В самом начале, в зародыше. Пресечь на корню какое-л. дело.
Правило гласит, что, если придаточная часть усечена до одного союзного слова, запятая перед ней не ставится. Д. Э. Розенталь, однако, замечает, что «в условиях контекста возможна постановка запятой и перед одиночным союзным словом». Судя по иллюстративным примерам, это происходит в случаях, когда союзное слово требуется логически подчеркнуть, «отталкиваясь» от имеющегося в контексте местоименного слова типа что или что-то, сравним: Что же надо делать? Научите, что; Одно время он что-то шептал, не могли понять — что? (А. Т.). Ваш пример аналогичен приведенным: Ксения понимала, что он на что-то намекал, но непонятно, на что.
Ответ был дан, по-видимому, по «Морфемно-орфографическому словарю» А. Н. Тихонова (М., 2002), то же в электронной версии словаря. В «Морфемно-словообразовательном словаре» И. В. Гурковой (М., 2012) представлены слова запасать и, как производное, запасаться, в которых обозначен корень запас-.
Однако более правильным представляется выделение в словах запас, запасать, запасти связного корня -пас-. Тот же корень в словах припасти, напасти, а при более этимологичном подходе еще и спасти, спасать. Такая точка зрения представлена, кроме упомянутых словарей Б. Т. Панова — А. В. Текучева и З. А. Потихи, в информационно-поисковой системе «Орфографическое комментирование русского словаря».
Более того, можно предположить, что в «Морфемно-орфографическом словаре» А. Н. Тихонова содержится опечатка или ошибка. Так, в статьях припасти, припастись и напасти, напастись дается отсылка к за/пас/ти, за/пас/ти/сь. В «Словообразовательном словаре русского языка» А. Н. Тихонова (М., 1990) статьи запасти, припасти, напасти снабжены перекрестными отсылками. Ср. статьи «Морфемно-орфографического словаря»:
запас | запа'с/
запасти | запас/ти'
напасти1 | на/пас/ти'1 [ср.: за/пас/ти']
напастись1 | на/пас/ти'/сь1 [ср.: за/пас/ти'/сь]
припасти | при/пас/ти' [ср.: за/пас/ти']
припастись | при/пас/ти'/сь [ср.: за/пас/ти'/сь]
Во второй части предложения есть формальное подлежащее – местоимение что, а значит, это двусоставное предложение. Получается, правило о непостановке запятой перед одиночными союзами и, или, либо, соединяющими структурно однотипные части, здесь не работает.
Вопрос об общем вводном слове в приведенном предложении не столь прост. С одной стороны, по смыслу слово наверное вполне может относиться как к первой грамматической основе (нездоровится), так и ко второй (что бывает), и это основание не ставить запятую перед союзом. С другой стороны, в предложении можно увидеть не предположения о том, что случилось (сравним действительно альтернативные версии: Наверное, ему нездоровится или его кто-то отвлёк), а сомнения в том, что слово нездоровится можно употребить по отношению к младенцу (в предложении не идет речь о двух предполагаемых ситуациях, а отражен поиск точного слова для обозначения одной ситуации). В таком случае вторая часть будет носить присоединительный характер, и это аргумент за постановку запятой перед союзом. В пользу этой версии может свидетельствовать то, что слово наверное невозможно присоединить напрямую ко второй части: *наверное, что там с младенцами бывает (тогда как предложение Наверное, ему нездоровится или его кто-то отвлёк легко раскладывается на два: Наверное, ему нездоровится и Наверное, его кто-то отвлёк).
Отметим, что приведенное предложение носит разговорный характер, а пунктуационные правила записи разговорной речи не разработаны детально, да и возможно ли их детально разработать, учтя все возможные варианты высказываний, – отдельный вопрос.
Первый вариант написания правильный.