commit 71d4c03c8d9b13f813a79c2578011d1331d7754d · aylac.top/advent_of_code

+28 -30
2015/18/rust/src/main.rs
···

       1
       1
        
       use std::mem::swap;

     

       2
       2
        
       

     

       3
       3
        
       fn generations(times: u32, mut world: Vec<u8>, size: usize, stuck: bool) -> Vec<u8> {

     

       4
       4
       -
           #[inline]

     

       5
       5
       -
           fn pos(x: usize, y: usize, size: usize) -> usize {

     

       6
       6
       -
               y * (size + 2) + x

     

       7
       7
       -
           }

     

       8
       8
       -
       

     

       9
       9
       -
           #[inline]

     

       10
       10
       -
           fn get_at(world: &Vec<u8>, size: usize, x: usize, y: usize) -> u8 {

     

       11
       11
       -
               // this is in known bounds

     

       12
       12
       -
               unsafe { *world.get_unchecked(pos(x, y, size)) }

     

       13
       13
       -
           }

     

       4
       4
       +
           let pos = |x, y| y * (size + 2) + x;

     

       14
       5
        
       

     

       15
       6
        
           let sizep = size + 1;

     

       16
       7
        
       

     

       17
       8
        
           let mut new_world = vec![0_u8; (size + 2).pow(2)];

     

       18
       9
        
       

     

       19
       19
       -
           if stuck {

     

       20
       20
       -
               world[pos(1, 1, size)] = 1;

     

       21
       21
       -
               world[pos(size, 1, size)] = 1;

     

       22
       22
       -
               world[pos(1, size, size)] = 1;

     

       23
       23
       -
               world[pos(size, size, size)] = 1;

     

       10
       10
       +
           unsafe {

     

       11
       11
       +
               if stuck {

     

       12
       12
       +
                   *world.get_unchecked_mut(pos(1, 1)) = 1;

     

       13
       13
       +
                   *world.get_unchecked_mut(pos(size, 1)) = 1;

     

       14
       14
       +
                   *world.get_unchecked_mut(pos(1, size)) = 1;

     

       15
       15
       +
                   *world.get_unchecked_mut(pos(size, size)) = 1;

     

       16
       16
       +
               }

     

       24
       17
        
           }

     

       25
       18
        
       

     

       26
       19
        
           for _ in 0..times {

     
···

       31
       24
        
                       let xm = xo - 1;

     

       32
       25
        
                       let xp = xo + 1;

     

       33
       26
        
       

     

       34
       34
       -
                       let was = get_at(&world, size, xo, yo) == 1;

     

       35
       35
       -
                       let neighbours = get_at(&world, size, xm, ym)

     

       36
       36
       -
                           + get_at(&world, size, xo, ym)

     

       37
       37
       -
                           + get_at(&world, size, xp, ym)

     

       38
       38
       -
                           + get_at(&world, size, xm, yo)

     

       39
       39
       -
                           + get_at(&world, size, xp, yo)

     

       40
       40
       -
                           + get_at(&world, size, xm, yp)

     

       41
       41
       -
                           + get_at(&world, size, xo, yp)

     

       42
       42
       -
                           + get_at(&world, size, xp, yp);

     

       43
       43
       -
                       new_world[pos(xo, yo, size)] = (neighbours == 3 || (neighbours == 2 && was)) as u8;

     

       27
       27
       +
                       unsafe {

     

       28
       28
       +
                           let was = *world.get_unchecked(pos(xo, yo)) == 1_u8;

     

       29
       29
       +
                           let neighbours = world.get_unchecked(pos(xm, ym))

     

       30
       30
       +
                               + world.get_unchecked(pos(xo, ym))

     

       31
       31
       +
                               + world.get_unchecked(pos(xp, ym))

     

       32
       32
       +
                               + world.get_unchecked(pos(xm, yo))

     

       33
       33
       +
                               + world.get_unchecked(pos(xp, yo))

     

       34
       34
       +
                               + world.get_unchecked(pos(xm, yp))

     

       35
       35
       +
                               + world.get_unchecked(pos(xo, yp))

     

       36
       36
       +
                               + world.get_unchecked(pos(xp, yp));

     

       37
       37
       +
                           *new_world.get_unchecked_mut(pos(xo, yo)) =

     

       38
       38
       +
                               (neighbours == 3 || (neighbours == 2 && was)) as u8;

     

       39
       39
       +
                       }

     

       44
       40
        
                   }

     

       45
       41
        
               }

     

       46
       42
        
       

     

       47
       43
        
               swap(&mut world, &mut new_world);

     

       48
       44
        
       

     

       49
       45
        
               // i hate the duplication here :(

     

       50
       50
       -
               if stuck {

     

       51
       51
       -
                   world[pos(1, 1, size)] = 1;

     

       52
       52
       -
                   world[pos(size, 1, size)] = 1;

     

       53
       53
       -
                   world[pos(1, size, size)] = 1;

     

       54
       54
       -
                   world[pos(size, size, size)] = 1;

     

       46
       46
       +
               unsafe {

     

       47
       47
       +
                   if stuck {

     

       48
       48
       +
                       *world.get_unchecked_mut(pos(1, 1)) = 1;

     

       49
       49
       +
                       *world.get_unchecked_mut(pos(size, 1)) = 1;

     

       50
       50
       +
                       *world.get_unchecked_mut(pos(1, size)) = 1;

     

       51
       51
       +
                       *world.get_unchecked_mut(pos(size, size)) = 1;

     

       52
       52
       +
                   }

     

       55
       53
        
               }

     

       56
       54
        
           }

     

       57
       55
        
           world