""" 0001.1 - Solution 1 - Brute Force Approach """
#####################################################################################
# Imports
#####################################################################################
from typing import List
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def twoSum(self, nums: List[int], target: int) -> List[int]:
"""Two Sum Function"""
for i in range(len(nums)):
for j in range(i + 1, len(nums)):
if nums[i] + nums[j] == target:
return [i, j]
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().twoSum([3, 3], 6))
print(Solution().twoSum([3, 2, 4], 6))
print(Solution().twoSum([2, 7, 11, 15], 9))
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()
""" 0001.1 - Solution 1 - Using Enumerate """
#####################################################################################
# Imports
#####################################################################################
from typing import List
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def twoSum(self, nums: List[int], target: int) -> List[int]:
"""Two Sum Function"""
for key1, num1 in enumerate(nums):
for key2, num2 in enumerate(nums[key1 + 1 :], key1 + 1):
# enumerate(iterable, start=1) -> start value determines starting index
# Printing value and key will output 1 i , 2 j, 3 k, 4 l
if num1 + num2 == target:
return [key1, key2]
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().twoSum([3, 3], 6))
print(Solution().twoSum([3, 2, 4], 6))
print(Solution().twoSum([2, 7, 11, 15], 9))
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()
""" 0495.1 - Teemo Attacking - Solution 1 - Linear Scan """
#####################################################################################
# Imports
#####################################################################################
from typing import List
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def findPoisonedDuration(self, timeSeries: List[int], duration: int) -> int:
"""Find Poisoned Duration Function"""
if not timeSeries:
return 0
total = 0
for i in range(len(timeSeries) - 1):
gap = timeSeries[i + 1] - timeSeries[i]
total += min(gap, duration)
total += duration # last attack always lasts full duration
return total
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().findPoisonedDuration([1, 4], 2)) # Expected: 4
print(Solution().findPoisonedDuration([1, 2], 2)) # Expected: 3
print(Solution().findPoisonedDuration([1, 2, 3, 4, 5], 5)) # Expected: 9
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()
""" 0495.2 - Teemo Attacking - Solution 2 - Linear Scan using zip() """
#####################################################################################
# Imports
#####################################################################################
from typing import List
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def findPoisonedDuration(self, timeSeries: List[int], duration: int) -> int:
"""Find Poisoned Duration Function"""
if not timeSeries:
return 0
total = 0
for curr, nxt in zip(timeSeries, timeSeries[1:]):
total += min(nxt - curr, duration)
total += duration # last attack always lasts full duration
return total
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().findPoisonedDuration([1, 4], 2)) # Expected: 4
print(Solution().findPoisonedDuration([1, 2], 2)) # Expected: 3
print(Solution().findPoisonedDuration([1, 2, 3, 4, 5], 5)) # Expected: 9
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()