Skip to content

Commit

Permalink
refactor: train composition remove intermediate variables
Browse files Browse the repository at this point in the history
  • Loading branch information
jdegand committed Dec 22, 2024
1 parent 2dfbafe commit e871959
Showing 1 changed file with 7 additions and 10 deletions.
17 changes: 7 additions & 10 deletions Java/TrainComposition.java
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,11 @@ public class TrainComposition {
// implementation same as my csharp solution

private LinkedList<Integer> train;

public TrainComposition()
{
train = new LinkedList<>();

public TrainComposition() {
train = new LinkedList<>();
}

public void attachWagonFromLeft(int wagonId) {
train.addFirst(wagonId);
}
Expand All @@ -19,20 +18,18 @@ public void attachWagonFromRight(int wagonId) {
}

public int detachWagonFromLeft() {
int node = train.removeFirst();
return node;
return train.removeFirst();
}

public int detachWagonFromRight() {
int node = train.removeLast();
return node;
return train.removeLast();
}

public static void main(String[] args) {
TrainComposition train = new TrainComposition();
train.attachWagonFromLeft(7);
train.attachWagonFromLeft(13);
System.out.println(train.detachWagonFromRight()); // 7
System.out.println(train.detachWagonFromRight()); // 7
System.out.println(train.detachWagonFromLeft()); // 13
}
}

0 comments on commit e871959

Please sign in to comment.