SQL 20250926

import%20marimo%0A%0A__generated_with%20%3D%20%220.17.6%22%0Aapp%20%3D%20marimo.App()%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_()%3A%0A%20%20%20%20import%20marimo%20as%20mo%0A%20%20%20%20return%20(mo%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20nav_menu%20%3D%20mo.nav_menu(%0A%20%20%20%20%20%20%20%20%7B%0A%20%20%20%20%20%20%20%20%20%20%20%20%22%2Findex.html%22%3A%20f%22%7Bmo.icon('lucide%3Ahome')%7D%20Home%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%22%2Fnotebooks%2FSQL_20250925.html%22%3A%20f%22%7Bmo.icon('lucide%3Aarrow-big-left')%7D%20Last%20Day%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%22%2Fnotebooks%2FSQL_20250929.html%22%3A%20f%22%7Bmo.icon('lucide%3Aarrow-big-right')%7D%20Next%20Day%22%2C%0A%20%20%20%20%20%20%20%20%7D%2C%0A%20%20%20%20%20%20%20%20orientation%3D%22horizontal%22%2C%0A%20%20%20%20)%0A%20%20%20%20nav_menu%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%20LeetCode%202783%0A%0A%20%20%20%20Table%3A%20Flights%0A%0A%20%20%20%20%7C%20Column%20Name%20%7C%20Type%20%7C%0A%20%20%20%20%7C-------------%7C------%7C%0A%20%20%20%20%7C%20flight_id%20%20%20%7C%20int%20%20%7C%0A%20%20%20%20%7C%20capacity%20%20%20%20%7C%20int%20%20%7C%0A%0A%20%20%20%20flight_id%20is%20the%20column%20with%20unique%20values%20for%20this%20table.%20Each%20row%20of%20this%20table%20contains%20flight%20id%20and%20its%20capacity.%0A%0A%20%20%20%20Table%3A%20Passengers%0A%0A%20%20%20%20%7C%20Column%20Name%20%20%7C%20Type%20%7C%0A%20%20%20%20%7C--------------%7C------%7C%0A%20%20%20%20%7C%20passenger_id%20%7C%20int%20%20%7C%0A%20%20%20%20%7C%20flight_id%20%20%20%20%7C%20int%20%20%7C%0A%0A%20%20%20%20passenger_id%20is%20the%20column%20with%20unique%20values%20for%20this%20table.%20Each%20row%20of%20this%20table%20contains%20passenger%20id%20and%20flight%20id.%20Passengers%20book%20tickets%20for%20flights%20in%20advance.%20If%20a%20passenger%20books%20a%20ticket%20for%20a%20flight%20and%20there%20are%20still%20empty%20seats%20available%20on%20the%20flight%2C%20the%20passenger%20ticket%20will%20be%20confirmed.%20However%2C%20the%20passenger%20will%20be%20on%20a%20waitlist%20if%20the%20flight%20is%20already%20at%20full%20capacity.%0A%0A%20%20%20%20Write%20a%20solution%20to%20report%20the%20number%20of%20passengers%20who%20successfully%20booked%20a%20flight%20(got%20a%20seat)%20and%20the%20number%20of%20passengers%20who%20are%20on%20the%20waitlist%20for%20each%20flight.%0A%0A%20%20%20%20Return%20the%20result%20table%20ordered%20by%20flight_id%20in%20ascending%20order.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20%22%22%22%0A%20%20%20%20WITH%20number_passenger%20AS%20(%0A%20%20%20%20%20%20%20%20SELECT%0A%20%20%20%20%20%20%20%20%20%20%20%20flight_id%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20COUNT(DISTINCT%20passenger_id)%20AS%20num_passenger%0A%20%20%20%20%20%20%20%20FROM%0A%20%20%20%20%20%20%20%20%20%20%20%20Passengers%0A%20%20%20%20%20%20%20%20GROUP%20BY%0A%20%20%20%20%20%20%20%20%20%20%20%201%0A%20%20%20%20)%0A%20%20%20%20SELECT%0A%20%20%20%20%20%20%20%20flight_id%2C%0A%20%20%20%20%20%20%20%20LEAST(COALESCE(num_passenger%2C0)%2C%20capacity)%20AS%20booked_cnt%2C%0A%20%20%20%20%20%20%20%20GREATEST(COALESCE(num_passenger%2C0)%20-%20capacity%2C%200)%20AS%20waitlist_cnt%0A%20%20%20%20FROM%0A%20%20%20%20%20%20%20%20Flights%0A%20%20%20%20%20%20%20%20%20%20%20%20LEFT%20JOIN%20number_passenger%20USING%20(flight_id)%0A%20%20%20%20ORDER%20BY%0A%20%20%20%20%20%20%20%201%3B%0A%20%20%20%20%22%22%22%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20import%20pandas%20as%20pd%0A%0A%20%20%20%20def%20waitlist_analysis(flights%3A%20pd.DataFrame%2C%20passengers%3A%20pd.DataFrame)%20-%3E%20pd.DataFrame%3A%0A%20%20%20%20%20%20%20%20number_passenger%20%3D%20passengers.groupby(%22flight_id%22%2C%20as_index%3DFalse)%5B%22passenger_id%22%5D.nunique()%0A%20%20%20%20%20%20%20%20df%20%3D%20pd.merge(flights%2C%20number_passenger%2C%20how%3D%22left%22%2C%20on%3D%22flight_id%22).fillna(0)%0A%20%20%20%20%20%20%20%20df%5B%22booked_cnt%22%5D%20%3D%20df%5B%5B%22capacity%22%2C%20%22passenger_id%22%5D%5D.min(axis%3D1)%0A%20%20%20%20%20%20%20%20df%5B%22waitlist_cnt%22%5D%20%3D%20(df%5B%22passenger_id%22%5D%20-%20df%5B%22capacity%22%5D).clip(lower%3D0)%0A%20%20%20%20%20%20%20%20df%20%3D%20df%5B%5B%22flight_id%22%2C%20%22booked_cnt%22%2C%20%22waitlist_cnt%22%5D%5D.sort_values(%22flight_id%22)%0A%20%20%20%20%20%20%20%20return%20df%0A%20%20%20%20return%0A%0A%0Aif%20__name__%20%3D%3D%20%22__main__%22%3A%0A%20%20%20%20app.run()%0A