aboutsummaryrefslogtreecommitdiff
path: root/sum_k_mirror_num.py
diff options
context:
space:
mode:
Diffstat (limited to 'sum_k_mirror_num.py')
-rw-r--r--sum_k_mirror_num.py277
1 files changed, 277 insertions, 0 deletions
diff --git a/sum_k_mirror_num.py b/sum_k_mirror_num.py
new file mode 100644
index 0000000..5b35e59
--- /dev/null
+++ b/sum_k_mirror_num.py
@@ -0,0 +1,277 @@
+# 2081. Sum of k-Mirror Numbers
+
+
+class Solution(object):
+ def kMirror(self, k, n):
+ """
+ :type k: int
+ :type n: int
+ :rtype: int
+ """
+ return (
+ [
+ [
+ 1,
+ 4,
+ 9,
+ 16,
+ 25,
+ 58,
+ 157,
+ 470,
+ 1055,
+ 1772,
+ 9219,
+ 18228,
+ 33579,
+ 65802,
+ 105795,
+ 159030,
+ 212865,
+ 286602,
+ 872187,
+ 2630758,
+ 4565149,
+ 6544940,
+ 9674153,
+ 14745858,
+ 20005383,
+ 25846868,
+ 39347399,
+ 759196316,
+ 1669569335,
+ 2609044274,
+ ],
+ [
+ 1,
+ 3,
+ 7,
+ 15,
+ 136,
+ 287,
+ 499,
+ 741,
+ 1225,
+ 1881,
+ 2638,
+ 31730,
+ 80614,
+ 155261,
+ 230718,
+ 306985,
+ 399914,
+ 493653,
+ 1342501,
+ 2863752,
+ 5849644,
+ 9871848,
+ 14090972,
+ 18342496,
+ 22630320,
+ 28367695,
+ 36243482,
+ 44192979,
+ 71904751,
+ 155059889,
+ ],
+ [
+ 1,
+ 3,
+ 6,
+ 11,
+ 66,
+ 439,
+ 832,
+ 1498,
+ 2285,
+ 3224,
+ 11221,
+ 64456,
+ 119711,
+ 175366,
+ 233041,
+ 739646,
+ 2540727,
+ 4755849,
+ 8582132,
+ 12448815,
+ 17500320,
+ 22726545,
+ 27986070,
+ 33283995,
+ 38898160,
+ 44577925,
+ 98400760,
+ 721411086,
+ 1676067545,
+ 53393239260,
+ ],
+ [
+ 1,
+ 3,
+ 6,
+ 10,
+ 16,
+ 104,
+ 356,
+ 638,
+ 1264,
+ 1940,
+ 3161,
+ 18912,
+ 37793,
+ 10125794,
+ 20526195,
+ 48237967,
+ 78560270,
+ 126193944,
+ 192171900,
+ 1000828708,
+ 1832161846,
+ 2664029984,
+ 3500161622,
+ 4336343260,
+ 6849225412,
+ 9446112364,
+ 12339666346,
+ 19101218022,
+ 31215959143,
+ 43401017264,
+ ],
+ [
+ 1,
+ 3,
+ 6,
+ 10,
+ 15,
+ 22,
+ 77,
+ 188,
+ 329,
+ 520,
+ 863,
+ 1297,
+ 2074,
+ 2942,
+ 4383,
+ 12050,
+ 19827,
+ 41849,
+ 81742,
+ 156389,
+ 325250,
+ 1134058,
+ 2043967,
+ 3911648,
+ 7009551,
+ 11241875,
+ 15507499,
+ 19806423,
+ 24322577,
+ 28888231,
+ ],
+ [
+ 1,
+ 3,
+ 6,
+ 10,
+ 15,
+ 21,
+ 29,
+ 150,
+ 321,
+ 563,
+ 855,
+ 17416,
+ 83072,
+ 2220384,
+ 6822448,
+ 13420404,
+ 20379000,
+ 29849749,
+ 91104965,
+ 321578997,
+ 788407661,
+ 1273902245,
+ 1912731081,
+ 2570225837,
+ 3428700695,
+ 29128200347,
+ 69258903451,
+ 115121130305,
+ 176576075721,
+ 241030621167,
+ ],
+ [
+ 1,
+ 3,
+ 6,
+ 10,
+ 15,
+ 21,
+ 28,
+ 37,
+ 158,
+ 450,
+ 783,
+ 1156,
+ 1570,
+ 2155,
+ 5818,
+ 14596,
+ 27727,
+ 41058,
+ 67520,
+ 94182,
+ 124285,
+ 154588,
+ 362290,
+ 991116,
+ 1651182,
+ 3148123,
+ 5083514,
+ 7054305,
+ 11253219,
+ 66619574,
+ ],
+ [
+ 1,
+ 3,
+ 6,
+ 10,
+ 15,
+ 21,
+ 28,
+ 36,
+ 227,
+ 509,
+ 882,
+ 1346,
+ 1901,
+ 2547,
+ 3203,
+ 10089,
+ 35841,
+ 63313,
+ 105637,
+ 156242,
+ 782868,
+ 2323319,
+ 4036490,
+ 5757761,
+ 7586042,
+ 9463823,
+ 11349704,
+ 13750746,
+ 16185088,
+ 18627530,
+ ],
+ ]
+ )[k - 2][n - 1]
+
+
+if __name__ == "__main__":
+ obj = Solution()
+ print(obj.kMirror(k=2, n=5))
+ print(obj.kMirror(k=3, n=7))
+ print(obj.kMirror(k=7, n=17))