Das deutsche Python-Forum

Code: Alles auswählen

   10 TI$="000000":R1=0:R2=0:OPEN 2,8,2,"INPUT02,S"
   20 IF ST<>0 THEN CLOSE 2:GOTO 500
   30 INPUT#2,L$
   40 A=ASC(MID$(L$,1,1))-65
   50 B=ASC(MID$(L$,3,1))-88
   60 GOSUB 700:R1=R1+O
   70 GOSUB 600:R2=R2+O:GOTO 20
  500 PRINT R1,R2:PRINT TI$:END
  600 ON B GOTO 630,640
  610 B=A-1:IF B<0 THEN B=2
  620 GOTO 700
  630 B=A:GOTO 700
  640 B=A+1:IF B=3 THEN B=0
  700 O=B+1
  710 IF A=B THEN O=O+3:RETURN
  720 IF A=0 AND B=1 OR A=1 AND B=2 OR A=2 AND B=0 THEN O=O+6
  730 RETURN

Code: Alles auswählen

print(sum({"A X":4+3j,"A Y":8+4j,"A Z":3+8j,"B X":1+1j,"B Y":5+5j,"B Z":9+9j,"C X":7+2j,"C Y":2+6j,"C Z":6+7j}[l.strip()]for l in open("i")))

Code: Alles auswählen

['GRRjbVjmJ', 'ZlgMRzzrN']

Code: Alles auswählen

#!/usr/bin/env python3
import string
from pathlib import Path

ITEMS_FILE = Path(r"C:\Users\Dennis\Documents\items.txt")


def read_file(file):
    with open(file, encoding="UTF-8") as items_file:
        return [line.strip("\n") for line in items_file]


def split_in_two_compartments(rucksacks):
    return [
        [compartments[: len(compartments) // 2], compartments[len(compartments) // 2 :]]
        for compartments in rucksacks
    ]


def find_common_items(rucksacks):
    common_items = []
    for rucksack in rucksacks:
        for item in rucksack[0]:
            if item in rucksack[1]:
                common_items.append(item)
    return common_items


def calculate_priority(items):
    priority = []
    for item in items:
        if item.islower():
            priority.append(string.ascii_lowercase.find(item) + 1)
        else:
            priority.append(string.ascii_uppercase.find(item) + 27)
    return sum(priority)


def main():
    rucksacks = read_file(ITEMS_FILE)
    rucksacks_with_split_compartments = split_in_two_compartments(rucksacks)
    common_items = find_common_items(rucksacks_with_split_compartments)
    print(calculate_priority(common_items))


if __name__ == "__main__":
    main()

Code: Alles auswählen

def calc(value):
    prio = ord(value) - 96
    if prio < 0: prio += 58
    return prio

def priority(sacks):
    packs = list(map(set, sacks))
    return calc((set.intersection(*packs)).pop())

data = open("input.txt").read().split("\n")[:-1]
mid = lambda x: len(x) // 2
data1 = [[pack[:mid(pack)], pack[mid(pack):]] for pack in data]
data2 = [data[i:i+3] for i in range(0, len(data), 3)]

print("Part 1:", sum(map(priority, data1)))
print("Part 2:", sum(map(priority, data2)))

Code: Alles auswählen

#!/usr/bin/env python3
import fileinput
from string import ascii_letters
from itertools import cycle

PRIOS = {char: i for i, char in enumerate(ascii_letters, 1)}


def iter_rucksacks():
    for rucksack in fileinput.input():
        rucksack = rucksack.rstrip()
        if rucksack:
            if len(rucksack) % 2:
                raise ValueError
            size = len(rucksack) // 2
            yield set(rucksack[:size]), set(rucksack[size:])


def main():
    sums = 0
    group_sums = 0
    for i, rucksack in zip(cycle(range(3)), iter_rucksacks()):
        left, right = rucksack
        common = left & right
        sums += sum(PRIOS[item] for item in common)
        if i == 0:
            group_badges = left | right
        else:
            group_badges &= left | right
        if len(group_badges) == 1:
            group_sums += PRIOS[group_badges.pop()]
    print(sums)
    print(group_sums)


if __name__ == "__main__":
    main()

Code: Alles auswählen

from string import ascii_letters as A
B=set(A).intersection
from more_itertools import chunked as C
r=list(map(str.strip,open("i")))
print(sum(A.find(B(*C(t,len(t)//2)).pop())+1for t in r),sum(A.find(B(*g).pop())+1for g in C(r,3)))

Code: Alles auswählen

   10 TI$="000000":R1=0:R2=0:DIM P(255),C(52),C2(52)
   20 P=1:FOR I=ASC("A") TO ASC("Z"):P(I)=P:P=P+1:NEXT
   30 FOR I=ASC("A") TO ASC("Z"):P(I)=P:P=P+1:NEXT
   40 FOR I=1 TO 52:C2(I)=1:NEXT
  100 OPEN 2,8,2,"INPUT03,S":LC=0
  110 IF ST<> 0 THEN CLOSE 2:GOTO 999
  120 FOR I=1 TO 52:C(I)=0:NEXT
  130 INPUT#2,L$:L=LEN(L$):M=L/2:LC=LC+1
  140 FOR I=1 TO M
  150 C(P(ASC(MID$(L$,I,1))))=1
  160 NEXT:FOR I=M+1 TO L
  170 P=P(ASC(MID$(L$,I,1)))
  180 IF C(P) THEN R1=R1+P:GOTO 200
  190 NEXT:STOP
  200 FOR I=M+1 TO L
  210 C(P(ASC(MID$(L$,I,1))))=1:NEXT
  220 FOR I=1 TO 52:C2(I)=C2(I) AND C(I):NEXT
  230 IF LC<>3 THEN 110
  240 FOR I=1 TO 52:IF C2(I) THEN R2=R2+I
  250 C2(I)=1:NEXT:LC=0:GOTO 110
  999 PRINT R1,R2:PRINT TI$

Code: Alles auswählen

10 T1!=TIMER:DEFINT A-Z:R1=0:R2=0:DIM P(255),C(52),C2(52)
20 P=1:FOR I=ASC("a") TO ASC("z"):P(I)=P:P=P+1:NEXT
30 FOR I=ASC("A") TO ASC("Z"):P(I)=P:P=P+1:NEXT
40 FOR I=1 TO 52:C2(I)=1:NEXT:DEF FN P(I)=P(ASC(MID$(L$,I,1)))
50 OPEN "input03.txt" FOR INPUT AS #1:LC=0
60 WHILE NOT EOF(1):LINE INPUT#1,L$:LC=LC+1:L=LEN(L$):M=L\2
70 FOR I=1 TO 52:C(I)=0:NEXT:FOR I=1 TO M:C(FN P(I))=1:NEXT
80 FOR I=M+1 TO L:P=FN P(I):IF C(P) THEN R1=R1+P:GOTO 100
90 NEXT:STOP:REM No common item error.
100 FOR I=M+1 TO L:C(FN P(I))=1:NEXT
110 FOR I=1 TO 52:C2(I)=C2(I) AND C(I):NEXT:IF LC<>3 THEN 135
120 FOR I=1 TO 52:IF C2(I) THEN R2=R2+I
130 C2(I)=1:NEXT:LC=0
135 WEND:CLOSE 1
140 PRINT R1,R2:PRINT"In";TIMER-T1!;"seconds."

Code: Alles auswählen

DECLARE FUNCTION GetPrio% (s AS STRING, i AS INTEGER)
DECLARE SUB InitItemSet (s() AS INTEGER, value AS INTEGER)
DECLARE SUB TickItems (s() AS INTEGER, L AS STRING, a AS INTEGER, b AS INTEGER)

t1 = TIMER
DIM SHARED Byte2Prio(255) AS INTEGER
DIM items(52) AS INTEGER, groupItems(52) AS INTEGER
DIM r1 AS INTEGER, r2 AS INTEGER, p AS INTEGER, lc AS INTEGER
DIM i AS INTEGER, length AS INTEGER, middle AS INTEGER

' Init lookup table ASCII code -> priority.
p = 1
FOR i = ASC("a") TO ASC("z"): Byte2Prio(i) = p: p = p + 1: NEXT
FOR i = ASC("A") TO ASC("Z"): Byte2Prio(i) = p: p = p + 1: NEXT

r1 = 0: r2 = 0: lc = 0
InitItemSet groupItems(), 1
OPEN "input03.txt" FOR INPUT AS #1
DO WHILE NOT EOF(1)
  LINE INPUT #1, L$: length = LEN(L$): middle = length \ 2
  ' Part 1
  InitItemSet items(), 0
  TickItems items(), L$, 1, middle
  FOR i = middle + 1 TO length
    p = GetPrio(L$, i): IF items(p) THEN r1 = r1 + p: EXIT FOR
  NEXT

  ' Part 2
  TickItems items(), L$, middle + 1, length
  FOR i = 1 TO 52: groupItems(i) = groupItems(i) AND items(i): NEXT
  lc = lc + 1
  IF lc = 3 THEN ' Group complete.
    lc = 0
    FOR i = 1 TO 52
      IF groupItems(i) THEN r2 = r2 + i: EXIT FOR
    NEXT
    InitItemSet groupItems(), 1
  END IF
LOOP
CLOSE 1: PRINT r1, r2: PRINT "In"; TIMER - t1; "seconds."

FUNCTION GetPrio% (s AS STRING, i AS INTEGER)
  GetPrio = Byte2Prio(ASC(MID$(s, i, 1)))
END FUNCTION

SUB InitItemSet (s() AS INTEGER, value AS INTEGER)
  DIM i AS INTEGER
  FOR i = 1 TO 52: s(i) = value: NEXT
END SUB

SUB TickItems (s() AS INTEGER, L AS STRING, a AS INTEGER, b AS INTEGER)
  DIM i AS INTEGER
  FOR i = a TO b: s(GetPrio(L, i)) = 1: NEXT
END SUB

Code: Alles auswählen

type
  TPrio = 1..52;
  TItems = set of TPrio;

const
  AllItems = [Low(TPrio)..High(TPrio)];

var
  charToPrioTable: array[0..255] of TPrio;
  r1, r2: Word;
  f: Text;
  lineCount: Byte;
  line: String;
  middle: Byte;
  items, groupItems: TItems;
  prio: TPrio;
  i: Byte;

procedure Init;
var
  i: Byte;
  p: TPrio;
begin
  FillChar(charToPrioTable, SizeOf(charToPrioTable), 0);
  p := 1;
  for i := Ord('a') to Ord('z') do
    begin
      charToPrioTable[i] := p;
      Inc(p);
    end;
  for i := Ord('A') to Ord('Z') do
    begin
      charToPrioTable[i] := p;
      Inc(p);
    end;
end;

function CharToPrio(c: Char): TPrio;
begin
  CharToPrio := charToPrioTable[Ord(c)];
end;

procedure IncludeItems(var items: TItems; const line: String; i, j: Byte);
begin
  for i := i to j do Include(items, CharToPrio(line[i]));
end;

begin
  Init;
  Assign(f, 'input03.txt');
  Reset(f);
  r1 := 0; r2 := 0; lineCount := 0; groupItems := AllItems;
  while not Eof(f) do
    begin
      ReadLn(f, line);
      middle := Length(line) DIV 2;
      items := [];
      { Part 1 }
      IncludeItems(items, line, 1, middle);
      for i := middle + 1 to Length(line) do
        begin
          prio := CharToPrio(line[i]);
          if prio in items then
            begin
              Inc(r1, prio);
              Break;
            end;
        end;

      { Part 2 }
      IncludeItems(items, line, middle + 1, Length(line));
      groupItems := groupItems * items;
      Inc(lineCount);
      if lineCount = 3 then
        begin
          lineCount := 0;
          for prio := Low(TPrio) to High(TPrio) do
            begin
              if prio in groupItems then
                begin
                  Inc(r2, prio);
                  Break;
                end;
            end;
        groupItems := AllItems;
        end;
    end;
  Close(f);
  WriteLn(r1, ' ', r2);
end.

Code: Alles auswählen

mid = lambda x: len(x) // 2
data1 = [[pack[:mid(pack)], pack[mid(pack):]] for pack in data]

Code: Alles auswählen

mid = lambda x: len(x) // 2
data1 = [[pack[:mid(pack)], pack[mid(pack):]] for pack in data]

Code: Alles auswählen

mid = len(pack) // 2
data1 = [[pack[:mid], pack[mid:]] for pack in data]

Code: Alles auswählen

data1 = [
    [pack[:mid], pack[mid:]] for pack in data
    for mid in (len(pack) // 2,)
]