""" 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()
""" 1588.1 - Sum of All Odd Length Subarrays - Solution 1 - Brute Force """
#####################################################################################
# Imports
#####################################################################################
from typing import List
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def sumOddLengthSubarrays(self, arr: List[int]) -> int:
"""Sum of All Odd Length Subarrays Function"""
total = 0
n = len(arr)
for i in range(n):
for length in range(1, n - i + 1, 2): # odd lengths only
total += sum(arr[i:i + length])
return total
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().sumOddLengthSubarrays([1, 4, 2, 5, 3])) # 58
print(Solution().sumOddLengthSubarrays([1, 2])) # 3
print(Solution().sumOddLengthSubarrays([10, 11, 12])) # 66
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()
""" 1588.2 - Sum of All Odd Length Subarrays - Solution 2 - Prefix Sum """
#####################################################################################
# Imports
#####################################################################################
from typing import List
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def sumOddLengthSubarrays(self, arr: List[int]) -> int:
"""Sum of All Odd Length Subarrays Function"""
n = len(arr)
prefix = [0] * (n + 1)
for i in range(n):
prefix[i + 1] = prefix[i] + arr[i]
total = 0
for i in range(n):
for length in range(1, n - i + 1, 2): # odd lengths only
total += prefix[i + length] - prefix[i]
return total
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().sumOddLengthSubarrays([1, 4, 2, 5, 3])) # 58
print(Solution().sumOddLengthSubarrays([1, 2])) # 3
print(Solution().sumOddLengthSubarrays([10, 11, 12])) # 66
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()
""" 1588.3 - Sum of All Odd Length Subarrays - Solution 3 - Math O(n) """
#####################################################################################
# Imports
#####################################################################################
from typing import List
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def sumOddLengthSubarrays(self, arr: List[int]) -> int:
"""Sum of All Odd Length Subarrays Function"""
total = 0
n = len(arr)
for i in range(n):
# Total subarrays containing arr[i]
total_subarrays = (i + 1) * (n - i)
# Odd-length subarrays containing arr[i]
odd_subarrays = (total_subarrays + 1) // 2
total += arr[i] * odd_subarrays
return total
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().sumOddLengthSubarrays([1, 4, 2, 5, 3])) # 58
print(Solution().sumOddLengthSubarrays([1, 2])) # 3
print(Solution().sumOddLengthSubarrays([10, 11, 12])) # 66
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()