solved #1329 with solution help

SouvikDcoder · SouvikDcoder · commit fe4d210c18bc · 2022-10-22T19:38:46.000+05:30
diff --git a/leetcode_kuro_RnD.ipynb b/leetcode_kuro_RnD.ipynb
@@ -9170,28 +9170,51 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 50,
+      "execution_count": 56,
       "metadata": {},
       "outputs": [
         {
           "name": "stdout",
           "output_type": "stream",
           "text": [
-            "[11, 25, 66, 1, 69, 7]\n",
-            "[23, 55, 17, 45, 15, 52]\n",
-            "[75, 31, 36, 44, 58, 8]\n",
-            "[22, 27, 33, 25, 68, 4]\n",
-            "[84, 28, 14, 11, 5, 50]\n"
+            "6 5\n",
+            "10\n",
+            "[[11, 25, 66, 1, 69, 7], [23, 55, 17, 45, 15, 52], [75, 31, 36, 44, 58, 8], [22, 27, 33, 25, 68, 4], [84, 28, 14, 11, 5, 50]]\n"
           ]
         }
       ],
       "source": [
         "# 1329 : https://leetcode.com/problems/sort-the-matrix-diagonally/\n",
         "\n",
+        "## ggot help from solution in comments : https://leetcode.com/problems/sort-the-matrix-diagonally/discuss/2659828/Easy-Solution-with\n",
+        "from collections import defaultdict\n",
+        "\n",
         "mat = [[11,25,66,1,69,7],[23,55,17,45,15,52],[75,31,36,44,58,8],[22,27,33,25,68,4],[84,28,14,11,5,50]]\n",
+        "col_mat = len(mat[0])\n",
+        "row_mat = len(mat)\n",
+        "\n",
+        "print(col_mat, row_mat)\n",
+        "\n",
+        "total_arrangements = (col_mat + row_mat) - 1\n",
+        "print(total_arrangements)\n",
+        "\n",
+        "diagonal_map = defaultdict(list)\n",
+        "for i in range(row_mat):\n",
+        "    for j in range(col_mat):\n",
+        "        diagonal_index = i - j\n",
+        "        element = mat[i][j]\n",
+        "        diagonal_map[diagonal_index].append(element)\n",
+        "    # print(diagonal_map)\n",
+        "\n",
+        "    for key in diagonal_map:\n",
+        "        diagonal_map[key].sort(reverse=True)\n",
+        "\n",
+        "    for j in range(row_mat):\n",
+        "        diagonal_index = i - j\n",
+        "        element = diagonal_map[diagonal_index].pop()\n",
+        "        mat[i][j] = element\n",
         "\n",
-        "for val in mat: \n",
-        "    print(val)"
+        "print(mat)\n"
       ]
     },
     {
