#1a)
mappingNumberKey = {}

ant_rekker = 6
ant_kol = 6

nr = 1

for i in range(ant_rekker):
    for j in range(ant_kol):
         mappingNumberKey[nr] = i,j
         nr += 1
            

#1b)
import random as ra

tall1 = ra.randint(1,ant_rekker*ant_kol)
tall2 = ra.randint(1,ant_rekker*ant_kol)

#Sikre oss at vi ikke trekker samme tallet for begge minene
while tall1 == tall2:
    tall2 = ra.randint(1,ant_rekker*ant_kol)
    

#1c)
import numpy as np


brett1 = np.zeros((ant_rekker,ant_kol))
brett2 = np.zeros((ant_rekker,ant_kol))

r1, c1 = mappingNumberKey[tall1]
r2, c2 = mappingNumberKey[tall2]

brett1[r1][c1] = np.nan
brett2[r2][c2] = np.nan

 
#1d) 
     
def finn_naboer(r_m, c_m, brett_m):
       
   for i in range(1,ant_rekker*ant_kol + 1):
      r, c = mappingNumberKey[i]
      dist = np.sqrt((r_m-r)**2 + (c_m-c)**2)
      if dist <= np.sqrt(2) and dist > 0:
         brett_m[r][c] = 1        
   
   return brett_m
     
brett1 = finn_naboer(r1, c1, brett1) 
brett2 = finn_naboer(r2, c2, brett2) 

brett_final = brett1+brett2
print(brett_final)