Jug 倒水问题 ZJU ACM

以前做过一个倒油的问题，和这道很像。

只是这里倒水可以往外倒，也可以往里接。假定外部可以倒出无数，也可以倒进无数。

最近比较忙，还没来得及格式化好输入输出。但算法就是下面这个了。

 

问题的链接

 

import java.util.ArrayList; import java.util.LinkedList; import java.util.Scanner; public class Main { public static void main(String[] args) { int[] input = { 3, 5, 4 }; Jug j = new Jug(input); j.RunJug(); } } class Jug { private int[] data; private ArrayList<JugState> StateList = new ArrayList<JugState>(); public Jug(int[] input) { this.data = input; } public void RunJug() { JugState state = new JugState(); state.waterInA = 0; state.waterInB = 0; this.StateList.add(state); FindNext(state); } private void FindNext(JugState state) { if (state.waterInA == data[2] || state.waterInB == data[2]) { this.PrintStateList(); System.out.println("success"); } else { if (CanPourOut(state)) { PourOut(state); } if (CanPourIn(state)) { PourIn(state); } // Pour A to B if (state.waterInB < data[1] && state.waterInA > 0) { PourAToB(state); } // Pour B to A if (state.waterInB > 0 && state.waterInA < data[0]) { PourBToA(state); } } } private boolean ContainState(JugState state) { for(int i=0;i<this.StateList.size();i++) { if(StateList.get(i).waterInA==state.waterInA&&StateList.get(i).waterInB==state.waterInB) return true; } return false; } private void PourBToA(JugState state) { JugState nextState = new JugState(); if (state.waterInB > data[0] - state.waterInA) { nextState.waterInA = data[0]; nextState.waterInB = state.waterInB - (data[0] - state.waterInA); } else { nextState.waterInB = 0; nextState.waterInA = state.waterInA + state.waterInB; } if (!ContainState(nextState)) { this.StateList.add(nextState); FindNext(nextState); this.StateList.remove(nextState); } } private void PourAToB(JugState state) { JugState nextState = new JugState(); if (state.waterInA > data[1] - state.waterInB) { nextState.waterInB = data[1]; nextState.waterInA = state.waterInA - (data[1] - state.waterInB); } else { nextState.waterInA = 0; nextState.waterInB = state.waterInB + state.waterInA; } if (!ContainState(nextState)) { this.StateList.add(nextState); FindNext(nextState); this.StateList.remove(nextState); } } private void PourIn(JugState state) { JugState nextState = new JugState(); if (state.waterInA < data[0]) { nextState.waterInA = data[0]; nextState.waterInB = state.waterInB; if (!ContainState(nextState)) { this.StateList.add(nextState); FindNext(nextState); this.StateList.remove(nextState); } } if (state.waterInB < data[1]) { nextState.waterInA = state.waterInA; nextState.waterInB = data[1]; if (!ContainState(nextState)) { this.StateList.add(nextState); FindNext(nextState); this.StateList.remove(nextState); } } } private boolean CanPourIn(JugState state) { if (state.waterInA < data[0] || state.waterInB < data[1]) return true; return false; } private void PourOut(JugState state) { JugState nextState = new JugState(); if (state.waterInA > 0) { nextState.waterInA = 0; nextState.waterInB = state.waterInB; if (!ContainState(nextState)) { this.StateList.add(nextState); FindNext(nextState); this.StateList.remove(nextState); } } if (state.waterInB > 0) { nextState.waterInA = state.waterInA; nextState.waterInB = 0; if (!ContainState(nextState)) { this.StateList.add(nextState); FindNext(nextState); this.StateList.remove(nextState); } } } private boolean CanPourOut(JugState state) { if (state.waterInA > 0 || state.waterInB > 0) return true; return false; } public void PrintStateList() { for(int i=0;i<this.StateList.size();i++) { System.out.println(this.StateList.get(i).waterInA+" , "+this.StateList.get(i).waterInB); } } public void PrintArray() { for (int i = 0; i < this.data.length; i++) { System.out.print(this.data[i]); } } } class JugState { int waterInA = 0; int waterInB = 0; }