Apartments

#!/usr/bin/env python3

'''Apartments'''

def main():
    '''Main'''
    n,m,k = [int(x) for x in input().split()]
    desired_sizes = [int(x) for x in input().split()]
    apartments_sizes = [int(x) for x in input().split()]
    apartments_sizes.sort()
    possible_apartments = [[] for _ in range(n)]
    for ind_apart, apart_size in enumerate(apartments_sizes):
        for ind_user, desired_size in enumerate(desired_sizes):
            if desired_size - k <= apart_size <= desired_size + k:
                possible_apartments[ind_user].append(ind_apart)
    possible_apartments.sort(key=len)
    cpt = 0
    index_apart_taken = []
    for elt in possible_apartments:
        for apart in elt:
            if apart not in index_apart_taken:
                cpt += 1
                index_apart_taken.append(apart)
                break
    print(cpt)

main()

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: TIME LIMIT EXCEEDED

Test 6

Verdict: TIME LIMIT EXCEEDED

Test 7

Verdict: TIME LIMIT EXCEEDED

Test 8

Verdict: TIME LIMIT EXCEEDED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: TIME LIMIT EXCEEDED

Test 14

Verdict: TIME LIMIT EXCEEDED

Test 15

Verdict: TIME LIMIT EXCEEDED

Test 16

Verdict: TIME LIMIT EXCEEDED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED