Unsolved/1635. Hopper Company Queries I (Hard).ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {
    "application/vnd.databricks.v1+cell": {
     "cellMetadata": {},
     "inputWidgets": {},
     "nuid": "f6ac9266-658c-4507-8a37-ecc81334b06d",
     "showTitle": false,
     "tableResultSettingsMap": {},
     "title": ""
    }
   },
   "source": [
    "## Importing Libraries"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "application/vnd.databricks.v1+cell": {
     "cellMetadata": {},
     "inputWidgets": {},
     "nuid": "d6194993-9312-4b46-8365-d6a73d8cb0b0",
     "showTitle": false,
     "tableResultSettingsMap": {},
     "title": ""
    }
   },
   "outputs": [],
   "source": [
    "from pyspark.sql.functions import *\n",
    "from pyspark.sql.types import *\n",
    "from pyspark.sql.window import Window"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "application/vnd.databricks.v1+cell": {
     "cellMetadata": {},
     "inputWidgets": {},
     "nuid": "23f6a680-8f01-4976-83fa-88103e49c6b8",
     "showTitle": false,
     "tableResultSettingsMap": {},
     "title": ""
    }
   },
   "source": [
    "**1635. Hopper Company Queries I (Hard)**\n",
    "\n",
    "**Table: Drivers**\n",
    "\n",
    "| Column Name | Type    |\n",
    "|-------------|---------|\n",
    "| driver_id   | int     |\n",
    "| join_date   | date    |\n",
    "\n",
    "driver_id is the primary key (column with unique values) for this table.\n",
    "Each row of this table contains the driver's ID and the date they joined the Hopper company.\n",
    " \n",
    "**Table: Rides**\n",
    "\n",
    "| Column Name  | Type    |\n",
    "|--------------|---------|\n",
    "| ride_id      | int     |\n",
    "| user_id      | int     |\n",
    "| requested_at | date    |\n",
    "\n",
    "ride_id is the primary key (column with unique values) for this table.\n",
    "Each row of this table contains the ID of a ride, the user's ID that requested it, and the day they requested it.\n",
    "There may be some ride requests in this table that were not accepted. \n",
    "\n",
    "**Table: AcceptedRides**\n",
    "\n",
    "| Column Name   | Type    |\n",
    "|---------------|---------|\n",
    "| ride_id       | int     |\n",
    "| driver_id     | int     |\n",
    "| ride_distance | int     |\n",
    "| ride_duration | int     |\n",
    "\n",
    "ride_id is the primary key (column with unique values) for this table.\n",
    "Each row of this table contains some information about an accepted ride.\n",
    "It is guaranteed that each accepted ride exists in the Rides table.\n",
    " \n",
    "**Write a solution to report the following statistics for each month of 2020:**\n",
    "- The number of drivers currently with the Hopper company by the end of the month (active_drivers).\n",
    "- The number of accepted rides in that month (accepted_rides).\n",
    "\n",
    "Return the result table ordered by month in ascending order, where month is the month's number (January is 1, February is 2, etc.).\n",
    "\n",
    "The result format is in the following example.\n",
    "\n",
    "**Example 1:**\n",
    "\n",
    "**Input:** \n",
    "\n",
    "**Drivers table:**\n",
    "| driver_id | join_date  |\n",
    "|-----------|------------|\n",
    "| 10        | 2019-12-10 |\n",
    "| 8         | 2020-1-13  |\n",
    "| 5         | 2020-2-16  |\n",
    "| 7         | 2020-3-8   |\n",
    "| 4         | 2020-5-17  |\n",
    "| 1         | 2020-10-24 |\n",
    "| 6         | 2021-1-5   |\n",
    "\n",
    "**Rides table:**\n",
    "| ride_id | user_id | requested_at |\n",
    "|---------|---------|--------------|\n",
    "| 6       | 75      | 2019-12-9    |\n",
    "| 1       | 54      | 2020-2-9     |\n",
    "| 10      | 63      | 2020-3-4     |\n",
    "| 19      | 39      | 2020-4-6     |\n",
    "| 3       | 41      | 2020-6-3     |\n",
    "| 13      | 52      | 2020-6-22    |\n",
    "| 7       | 69      | 2020-7-16    |\n",
    "| 17      | 70      | 2020-8-25    |\n",
    "| 20      | 81      | 2020-11-2    |\n",
    "| 5       | 57      | 2020-11-9    |\n",
    "| 2       | 42      | 2020-12-9    |\n",
    "| 11      | 68      | 2021-1-11    |\n",
    "| 15      | 32      | 2021-1-17    |\n",
    "| 12      | 11      | 2021-1-19    |\n",
    "| 14      | 18      | 2021-1-27    |\n",
    "\n",
    "**AcceptedRides table:**\n",
    "| ride_id | driver_id | ride_distance | ride_duration |\n",
    "|---------|-----------|---------------|---------------|\n",
    "| 10      | 10        | 63            | 38            |\n",
    "| 13      | 10        | 73            | 96            |\n",
    "| 7       | 8         | 100           | 28            |\n",
    "| 17      | 7         | 119           | 68            |\n",
    "| 20      | 1         | 121           | 92            |\n",
    "| 5       | 7         | 42            | 101           |\n",
    "| 2       | 4         | 6             | 38            |\n",
    "| 11      | 8         | 37            | 43            |\n",
    "| 15      | 8         | 108           | 82            |\n",
    "| 12      | 8         | 38            | 34            |\n",
    "| 14      | 1         | 90            | 74            |\n",
    "\n",
    "**Output:** \n",
    "| month | active_drivers | accepted_rides |\n",
    "|-------|----------------|----------------|\n",
    "| 1     | 2              | 0              |\n",
    "| 2     | 3              | 0              |\n",
    "| 3     | 4              | 1              |\n",
    "| 4     | 4              | 0              |\n",
    "| 5     | 5              | 0              |\n",
    "| 6     | 5              | 1              |\n",
    "| 7     | 5              | 1              |\n",
    "| 8     | 5              | 1              |\n",
    "| 9     | 5              | 0              |\n",
    "| 10    | 6              | 0              |\n",
    "| 11    | 6              | 2              |\n",
    "| 12    | 6              | 1              |\n",
    "\n",
    "**Explanation:** \n",
    "- By the end of January --> two active drivers (10, 8) and no accepted rides.\n",
    "- By the end of February --> three active drivers (10, 8, 5) and no accepted rides.\n",
    "- By the end of March --> four active drivers (10, 8, 5, 7) and one accepted ride (10).\n",
    "- By the end of April --> four active drivers (10, 8, 5, 7) and no accepted rides.\n",
    "- By the end of May --> five active drivers (10, 8, 5, 7, 4) and no accepted rides.\n",
    "- By the end of June --> five active drivers (10, 8, 5, 7, 4) and one accepted ride (13).\n",
    "- By the end of July --> five active drivers (10, 8, 5, 7, 4) and one accepted ride (7).\n",
    "- By the end of August --> five active drivers (10, 8, 5, 7, 4) and one accepted ride (17).\n",
    "- By the end of September --> five active drivers (10, 8, 5, 7, 4) and no accepted rides.\n",
    "- By the end of October --> six active drivers (10, 8, 5, 7, 4, 1) and no accepted rides.\n",
    "- By the end of November --> six active drivers (10, 8, 5, 7, 4, 1) and two accepted rides (20, 5).\n",
    "- By the end of December --> six active drivers (10, 8, 5, 7, 4, 1) and one accepted ride (2)."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "application/vnd.databricks.v1+cell": {
     "cellMetadata": {},
     "inputWidgets": {},
     "nuid": "a762f84d-f622-4403-b695-4066e5bdbaf0",
     "showTitle": false,
     "tableResultSettingsMap": {},
     "title": ""
    }
   },
   "outputs": [],
   "source": [
    "drivers_data_1635 = [\n",
    "    (10, \"2019-12-10\"), (8, \"2020-01-13\"), (5, \"2020-02-16\"), \n",
    "    (7, \"2020-03-08\"), (4, \"2020-05-17\"), (1, \"2020-10-24\"), \n",
    "    (6, \"2021-01-05\")\n",
    "]\n",
    "\n",
    "drivers_columns_1635 = [\"driver_id\", \"join_date\"]\n",
    "drivers_df_1635 = spark.createDataFrame(drivers_data_1635, drivers_columns_1635)\n",
    "drivers_df_1635.show()\n",
    "\n",
    "rides_data_1635 = [\n",
    "    (6, 75, \"2019-12-09\"), (1, 54, \"2020-02-09\"), (10, 63, \"2020-03-04\"),\n",
    "    (19, 39, \"2020-04-06\"), (3, 41, \"2020-06-03\"), (13, 52, \"2020-06-22\"),\n",
    "    (7, 69, \"2020-07-16\"), (17, 70, \"2020-08-25\"), (20, 81, \"2020-11-02\"),\n",
    "    (5, 57, \"2020-11-09\"), (2, 42, \"2020-12-09\"), (11, 68, \"2021-01-11\"),\n",
    "    (15, 32, \"2021-01-17\"), (12, 11, \"2021-01-19\"), (14, 18, \"2021-01-27\")\n",
    "]\n",
    "\n",
    "rides_columns_1635 = [\"ride_id\", \"user_id\", \"requested_at\"]\n",
    "rides_df_1635 = spark.createDataFrame(rides_data_1635, rides_columns_1635)\n",
    "rides_df_1635.show()\n",
    "\n",
    "accepted_rides_data_1635 = [\n",
    "    (10, 10, 63, 38), (13, 10, 73, 96), (7, 8, 100, 28), (17, 7, 119, 68),\n",
    "    (20, 1, 121, 92), (5, 7, 42, 101), (2, 4, 6, 38), (11, 8, 37, 43),\n",
    "    (15, 8, 108, 82), (12, 8, 38, 34), (14, 1, 90, 74)\n",
    "]\n",
    "\n",
    "accepted_rides_columns_1635 = [\"ride_id\", \"driver_id\", \"ride_distance\", \"ride_duration\"]\n",
    "accepted_rides_df_1635 = spark.createDataFrame(accepted_rides_data_1635, accepted_rides_columns_1635)\n",
    "accepted_rides_df_1635.show()"
   ]
  }
 ],
 "metadata": {
  "application/vnd.databricks.v1+notebook": {
   "computePreferences": null,
   "dashboards": [],
   "environmentMetadata": {
    "base_environment": "",
    "environment_version": "1"
   },
   "inputWidgetPreferences": null,
   "language": "python",
   "notebookMetadata": {
    "pythonIndentUnit": 4
   },
   "notebookName": "1635. Hopper Company Queries I (Hard)",
   "widgets": {}
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}