From 5498f566a0bc1f2364f735cd040a129a94c0434e Mon Sep 17 00:00:00 2001 From: Mika Suhonen Date: Tue, 6 Dec 2022 12:46:07 +0200 Subject: [PATCH] Function for finding unique string --- 06/f.py | 24 ++++++++++-------------- 1 file changed, 10 insertions(+), 14 deletions(-) diff --git a/06/f.py b/06/f.py index c075e3f..36641d9 100644 --- a/06/f.py +++ b/06/f.py @@ -6,19 +6,15 @@ input = open(os.path.dirname(__file__) + index, index1, index2 = 0, 0, 0 buf = [] -for x in input: - index += 1 - buf.append(x) - if index1 == 0 and len(buf) == 4: - if len(set(buf)) == 4: - index1 = index - buf.pop(0) - if index2 == 0 and len(buf) == 14: - if len(set(buf)) == 14: - index2 = index - break - buf.pop(0) +def findUniq(length): + ind = 0 + for x in input: + ind += 1 + if len(set(input[ind:ind+length])) == length: + return ind+length + return False -print("First answer: " + str(index1)) -print("Second answer: " + str(index2)) + +print("First answer: " + str(findUniq(4))) +print("Second answer: " + str(findUniq(14)))