adventofcode2015/Day22.java at master · nsiwnf/adventofcode2015 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140


import java.util.Stack;

public class Day22 {

  private static final int MISSILE = 53;

  private static final int DRAIN = 73;

  private static final int SHIELD = 113;

  private static final int POISON = 173;

  private static final int RECHARGE = 229;

  private static int part1() {
    int minSpent = Integer.MAX_VALUE;

    Stack<Point> points = new Stack<>();
    Point start = new Point(55, 50, 500, 0);
    // Point start = new Point(13, 10, 250, 0);
    start.playerTurn = true;
    points.add(start);

    while (!points.isEmpty()) {
      Point p = points.pop();

      p.applyEffects();

      if (p.won()) {
        minSpent = Math.min(minSpent, p.manaSpent);
      }
      else if (p.hp > 0 && p.manaSpent < minSpent) {
        if (p.playerTurn) {
          if (p.mana > MISSILE) {
            points.push(
                new Point(p.bossHp - 4, p.hp, p.mana - MISSILE, p.shield, p.poison, p.recharge, p.manaSpent + MISSILE));
          }
          if (p.mana > DRAIN) {
            points.push(
                new Point(p.bossHp - 2, p.hp + 2, p.mana - DRAIN, p.shield, p.poison, p.recharge, p.manaSpent + DRAIN));
          }
          if (p.mana > SHIELD && p.shield == 0) {
            points.push(new Point(p.bossHp, p.hp, p.mana - SHIELD, 6, p.poison, p.recharge, p.manaSpent + SHIELD));
          }
          if (p.mana > POISON && p.poison == 0) {
            points.push(new Point(p.bossHp, p.hp, p.mana - POISON, p.shield, 6, p.recharge, p.manaSpent + POISON));
          }
          if (p.mana > RECHARGE && p.recharge == 0) {
            points.push(new Point(p.bossHp, p.hp, p.mana - RECHARGE, p.shield, p.poison, 5, p.manaSpent + RECHARGE));
          }
        }
        else {
          p.hp -= 8;
          p.playerTurn = true;
          points.push(p);
        }
      }

    }

    return minSpent;
  }

  static class Point {
    boolean playerTurn = false;

    int bossHp;

    int hp;

    int mana;

    int manaSpent;

    // counters
    int shield;

    int poison;

    int recharge;

    Point(int bossHp, int hp, int mana, int manaSpent) {
      this.bossHp = bossHp;
      this.hp = hp;
      this.mana = mana;
      this.manaSpent = manaSpent;
    }

    Point(int bossHp, int hp, int mana, int shield, int poison, int recharge, int manaSpent) {
      this.bossHp = bossHp;
      this.hp = hp;
      this.mana = mana;
      this.shield = shield;
      this.poison = poison;
      this.recharge = recharge;
      this.manaSpent = manaSpent;
    }

    boolean won() {
      return bossHp <= 0 && hp > 0;
    }

    void applyEffects() {
      if(playerTurn) {
        hp--;
      }
      if (bossHp > 0 && hp > 0) {
        if (shield > 0) {
          if (!playerTurn && hp > 1)
            hp += 7;
          shield--;
        }

        if (poison > 0) {
          bossHp -= 3;
          poison--;
        }

        if (recharge > 0) {
          mana += 101;
          recharge--;
        }
      }
    }
  }

  public static void main(String[] args) {
    // BOSS
    // Hit Points: 55
    // Damage: 8

    // YOU
    // HP 50
    // Mana 500
    System.out.println(part1()); // 953

  }
}