aboutsummaryrefslogtreecommitdiff
path: root/maximum_swap.py
diff options
context:
space:
mode:
Diffstat (limited to 'maximum_swap.py')
-rw-r--r--maximum_swap.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/maximum_swap.py b/maximum_swap.py
new file mode 100644
index 0000000..ce384c1
--- /dev/null
+++ b/maximum_swap.py
@@ -0,0 +1,30 @@
+# 670. Maximum Swap
+
+"""
+you are given an integer 'num'. you can swap two digits at most once to get
+the maximum valued number. return the maximum valued number you can get.
+"""
+
+
+class Solution(object):
+ def maximumSwap(self, num):
+ """
+ :type num: int
+ :rtype: int
+ """
+ num = [int(x) for x in str(num)]
+ max_idx, xi, yi = len(num) - 1, 0, 0
+ for i in range(len(num) - 1, -1, -1):
+ if num[i] > num[max_idx]:
+ max_idx = i
+ elif num[i] < num[max_idx]:
+ xi = i
+ yi = max_idx
+ num[xi], num[yi] = num[yi], num[xi]
+ return int("".join([str(x) for x in num]))
+
+
+if __name__ == "__main__":
+ obj = Solution()
+ print(obj.maximumSwap(2736))
+ print(obj.maximumSwap(9973))