הסכום הגדול ביותר

המורה שלך כתב על הלוח שלושה מספרים חיוביים: A, B, C. לאחר מכן, המורה מציב בפניך אתגר: בחר אחד משלושת המספרים והחלף אותו בריבוע שלו (X²), כך שסכום שלושת המספרים לאחר ההחלפה יהיה הגדול ביותר האפשרי.

כתוב פונקציה בשם sum שמקבלת שלושה מספרים טבעיים A, B, C מהקלט. החזר את הסכום הגדול ביותר האפשרי לאחר החלפת מספר אחד בריבוע שלו.

קלט
1 5 5

פלט
31

קלט
0 1 10

פלט
101

הסבר #1: אנחנו מחליפים אחד מהמספרים 5 בריבוע שלו — 25, ולכן מקבלים 1 + 5 + 25 = 31
הסבר #2: אנחנו מחליפים את 10 בריבוע שלו — 100, ומקבלים 0 + 1 + 100 = 101