/*📌 3️⃣ Shipment & Package Management System
Problem Statement:
Design a Shipment & Package Management System with the following features:
1.  Add a package to a shipment (addPackage(packageId, weight, destination)).
2.  Assign packages to shipments (assignShipment(shipmentId)).
3.  Calculate total shipment cost based on package weight & distance.
4.  Track shipment status (Pending, In Transit, Delivered).
🔹 Constraints:
✅ Each shipment has a weight limit of 100 kg.
✅ Packages should be prioritized based on weight (heavier first).
✅ Cost = Base Fee ($10) + $2 per kg + $5 per 100 miles.
*/
import java.util.*;
class ShipmentSystem
{
  PriorityQueue<int[]> pq=new PriorityQueue<>((a,b)->b[1]-a[1]);
  HashMap<Integer,List<int[]>> shipments=new HashMap<>();
  HashMap<Integer,String> shipmentStatus=new HashMap<>();

  public void addPackage(int packageId, int weight, int distance)
  {
    pq.add(new int[]{packageId, weight, distance});
  }

  public void assignShipment(int shipmentId)
  {
    if (shipments.containsKey(shipmentId)) {
      throw new IllegalArgumentException("Shipment ID already exists!");
    }
    int weight=0;
    List<int[]> tmp=new ArrayList<>();
    while(!pq.isEmpty() && weight+pq.peek()[1]<=100)
    {
      int[] pkg=pq.remove();
      tmp.add(pkg);
      weight=weight+pkg[1];
    }
    shipments.put(shipmentId, tmp);
    shipmentStatus.put(shipmentId, "Pending");
  }

  public void calculateShipmentCost(int shipmentId)     //based on pkg weight,distance
  {
    if(!shipments.containsKey(shipmentId))
      throw new IllegalArgumentException("Invalid shipment id!");
    List<int[]> shipmentList=shipments.get(shipmentId);
    double totalCost=0;
    for(int[] pkg:shipmentList)
    {
      double cost=pkg[1]*2+(5*pkg[2]/100);
      totalCost+=cost;
    }   
    totalCost+=10;
    System.out.println("Cost of shipment id: "+shipmentId+" = "+totalCost);
  }
  
  public void trackShipment(int shipmentId)    //(Pending, In Transit, Delivered)
  {
    if(!shipmentStatus.containsKey(shipmentId))
      throw new IllegalArgumentException("Invalid shipment id!");
    System.out.println(shipmentStatus.get(shipmentId));
  }

  public void updateShipmentStatus(int shipmentId, String status) {
      if (!shipmentStatus.containsKey(shipmentId))
          throw new IllegalArgumentException("Invalid shipment id!");
      
      shipmentStatus.put(shipmentId, status);
  }
}


class Solution
{
  public static void main(String[] args)
  {
    ShipmentSystem obj=new ShipmentSystem();
    obj.addPackage(1, 30, 500);
    obj.addPackage(2, 50, 1200);
    obj.addPackage(3, 20, 800);
    obj.assignShipment(101);
    obj.trackShipment(101);
    obj.updateShipmentStatus(101, "In Transit");
    obj.trackShipment(101);
    obj.calculateShipmentCost(101);
    obj.updateShipmentStatus(101, "Delivered");
    obj.trackShipment(101);
  }

}