[JAVA][ZOJ 1004][Anagrams by Stack]

[java] view
plaincopyprint?

1. import java.io.BufferedInputStream;
2. import java.util.Scanner;
3. public class Main {
4. static String start;// record the first str
5. static String end;// record the rearranged str
6. public static void dfs(char[] stack, char[] sequence, int posInStart, int posInEnd, int posInStack) {
7. if (posInStart + posInEnd == start.length() * 2) {// if all the letter has been rearranged
8. for (int i = 0; i < sequence.length; i++) {
9. System.out.print(sequence[i] + " ");// output the sequence
10. }
11. System.out.println();
12. }
13. if (posInStart < start.length()) {
14. sequence[posInStart + posInEnd] = ‘i‘;
15. stack[++posInStack] = start.charAt(posInStart);// push in
16. dfs(stack, sequence, posInStart + 1, posInEnd, posInStack);
17. posInStack--;// pop out
18. }
19. if (posInStack >= 0 && posInEnd < end.length() && stack[posInStack] == end.charAt(posInEnd)) {// pop out
20. sequence[posInStart + posInEnd] = ‘o‘;
21. posInStack--;// pop out
22. dfs(stack, sequence, posInStart, posInEnd + 1, posInStack);
23. stack[++posInStack] = end.charAt(posInEnd);
24. }
25. }
26. public static void main(String[] args) {
27. Scanner sc = new Scanner(new BufferedInputStream(System.in));
28. while (sc.hasNext()) {
29. start = sc.next();
30. end = sc.next();
31. char[] stack = new char[start.length() * 2];// use array as the stack
32. char[] sequence = new char[start.length() * 2];// record the operation seq, it will have (start.length() * 2) times operations
33. System.out.println("[");
34. dfs(stack, sequence, 0, 0, -1);
35. System.out.println("]");
36. }
37. }
38. }

最近写的几道题都用到了dfs，用数组代替栈的使用，控制好stack数字下标就没什么大问题，难得一次AC