یک ذره با جرم m از ارتفاع h رها می‌شود. بر اساس قانون حفظ انرژی، می‌توان ارتفاع کلی ذره را محاسبه کرد. نوشتن تابعی که جرم و ارتفاع ذره را گرفته و ارتفاع کلی که ذره به سطح زمین خواهد رسید را محاسبه کند. برای محاسبه می‌توانید از انرژی جنبشی و جابجایی استفاده کنید. فرض کنید ثقل را در نظر نمی‌گیریم و اصطلاحا اصل انرژی حفظ می‌شود.
\n
توجه: برای ایجاد راحتی در محاسبات، ارتفاع به واحد متر و جرم به واحد کیلوگرم ورودی داده می‌شوند.
\n
تابعی بنویسید که جرم و ارتفاع ذره را به عنوان ورودی بگیرد و ارتفاع کلی که ذره به سطح زمین خواهد رسید را در خروجی برگرداند.
\n
نمونه ورودی و خروجی
\n
total_height(5, 10) ➞ 10
\n
(total_height(1, 2)) ➞ 2
\n
(total_height(0.5, 100)) ➞ 100

total_height(5, 10)  ➞ 10

total_height(1, 2)  ➞ 2

total_height(0.5, 100)  ➞ 100