201652026-01-03 16:46:58marnorlorOsztozkodás (75 pont)python3Accepted 75/75126ms14884 KiB
def osztozas(N: int, K: int, A: list):
    ajandekA=[]
    ajandekB=[]
    max2=0
    kOsszeg=0
    osszegA=0
    osszegB=0
    j=0
    A.sort(reverse=True)
    for i in range(j, len(A)):
        kOsszeg=kOsszeg+A[i]
    for j in range(0, len(A)):
        max2=A[j]
        if len(A)-j<=K:
            if kOsszeg>0:
                if j % 2==0:
                    osszegA=osszegA+kOsszeg
                if j % 2==1:
                    osszegB=osszegB+kOsszeg
                break
            else:
                if j % 2==0:
                    osszegA=osszegA+max2
                if j % 2==1:
                    osszegB=osszegB+max2
        else:
            if j % 2==0:
                    osszegA=osszegA+max2
            if j % 2==1:
                    osszegB=osszegB+max2
        kOsszeg=kOsszeg-A[j]
    return osszegA, osszegB

def test():
    a, b=osztozas(5, 3, [-2, 3, -1, 4, 5])
    print(a, b)
    assert 6==a
    assert 3==b

    a, b=osztozas(5, 3, [1, 4, 2, 4, 5])
    print(a, b)
    assert 12==a
    assert 4==b  
    a, b=osztozas(5, 3, [2, 3, 1, 5, 4])
    assert 11==a
    assert 4==b

    a, b=osztozas(5, 0, [2, 3, 1, 5, 4])
    assert 9==a
    assert 6==b

    a, b=osztozas(2, 2, [-5, 3])
    assert 3==a
    assert -5==b

    N=100000
    E=[0]*N
    for i in range(0, N):
        E[i]=i
    a, b=osztozas(N, 0, E)
    assert 2500000000==a
    assert 2499950000==b
    
    a, b=osztozas(20, 5, [7, 8, -4, 3, 10, 7, 4, 10, 6, 1, -1, 4, -1, -2, 3, -1, 4, -2, -3, 5])
    assert 32==a
    assert 26==b

def futtatas():
    N, K=map(int,input().split())
    A=list(map(int,input().split()))
    a, b=osztozas(N, K, A)
    print(a, b)

#test()
futtatas()
SubtaskSumTestVerdictTimeMemory
base75/75
1Accepted0/017ms3364 KiB
2Accepted0/016ms3332 KiB
3Accepted2/217ms3124 KiB
4Accepted3/317ms3124 KiB
5Accepted3/317ms3352 KiB
6Accepted2/217ms3556 KiB
7Accepted3/318ms3580 KiB
8Accepted2/218ms3584 KiB
9Accepted3/319ms3544 KiB
10Accepted2/218ms3504 KiB
11Accepted3/317ms3592 KiB
12Accepted1/1111ms14288 KiB
13Accepted2/2114ms14176 KiB
14Accepted1/1105ms14172 KiB
15Accepted2/286ms14200 KiB
16Accepted1/1104ms13620 KiB
17Accepted2/2108ms14884 KiB
18Accepted1/198ms13752 KiB
19Accepted2/275ms11712 KiB
20Accepted2/275ms11756 KiB
21Accepted3/3115ms14848 KiB
22Accepted2/2107ms14764 KiB
23Accepted3/3111ms14828 KiB
24Accepted2/2115ms14780 KiB
25Accepted3/3115ms14772 KiB
26Accepted2/2107ms14844 KiB
27Accepted3/3118ms14168 KiB
28Accepted2/2126ms14132 KiB
29Accepted3/3116ms14260 KiB
30Accepted3/316ms3124 KiB
31Accepted3/316ms3124 KiB
32Accepted3/316ms3124 KiB
33Accepted3/316ms3276 KiB
34Accepted3/316ms3340 KiB