Python & Swift Code for Determining if a Number Can Be Divided into Three Equal Parts

This code snippet addresses a problem where you need to determine if a given number 'n' can be divided into three equal parts. The code takes additional parameters 'k', 'd1', and 'd2' as input and evaluates several conditions to arrive at a 'yes' or 'no' answer. The provided code examples are presented in both Python and Swift for a comparative analysis.

Python Code

import sys
T=int(sys.stdin.readline().strip())
for t in range(T):
    n,k,d1,d2=map(int,sys.stdin.readline().strip().split())
    if n%3!=0:
        print('no')
    elif k-d1-d2>=0 and (k-d1-d2)%3==0 and n-k-max(d1,d2)-abs(d1-d2)>=0 and (n-k-max(d1,d2)-abs(d1-d2))%3==0:
        print('yes')
    elif k-2*d1-d2>=0 and (k-2*d1-d2)%3==0 and n-k-d1-2*d2>=0 and (n-k-d1-2*d2)%3==0:
        print('yes')
    elif k-d1-2*d2>=0 and (k-d1-2*d2)%3==0 and n-k-2*d1-d2>=0 and (n-k-2*d1-d2)%3==0:
        print('yes')
    elif k-max(d1,d2)-abs(d1-d2)>=0 and (k-max(d1,d2)-abs(d1-d2))%3==0 and n-k-d1-d2>=0 and (n-k-d1-d2)%3==0:
        print('yes')
    else:
        print('no')

Swift Code

T = Int(readLine()!)!

for _ in 0..<T {
    let inputs = readLine()!.split(separator: " ").map { Int($0)! }
    let n = inputs[0]
    let k = inputs[1]
    let d1 = inputs[2]
    let d2 = inputs[3]
    
    if n % 3 != 0 {
        print("no")
    } else if k - d1 - d2 >= 0 && (k - d1 - d2) % 3 == 0 && n - k - max(d1, d2) - abs(d1 - d2) >= 0 && (n - k - max(d1, d2) - abs(d1 - d2)) % 3 == 0 {
        print("yes")
    } else if k - 2 * d1 - d2 >= 0 && (k - 2 * d1 - d2) % 3 == 0 && n - k - d1 - 2 * d2 >= 0 && (n - k - d1 - 2 * d2) % 3 == 0 {
        print("yes")
    } else if k - d1 - 2 * d2 >= 0 && (k - d1 - 2 * d2) % 3 == 0 && n - k - 2 * d1 - d2 >= 0 && (n - k - 2 * d1 - d2) % 3 == 0 {
        print("yes")
    } else if k - max(d1, d2) - abs(d1 - d2) >= 0 && (k - max(d1, d2) - abs(d1 - d2)) % 3 == 0 && n - k - d1 - d2 >= 0 && (n - k - d1 - d2) % 3 == 0 {
        print("yes")
    } else {
        print("no")
    }
}

This code demonstrates a practical application of logic and arithmetic in programming. It highlights the ability of code to solve complex problems by systematically checking conditions and evaluating expressions. The use of both Python and Swift provides a comparative insight into the syntax and structure of these popular programming languages. If you're interested in further exploration of similar problems involving division or number manipulation, you can find numerous resources online and in programming textbooks.