|
1 | 1 | classSolution { |
| 2 | + |
| 3 | +privatestaticfinalintSWAP_DONE_IDX =Integer.MIN_VALUE; |
| 4 | + |
2 | 5 | publicbooleanbuddyStrings(Strings,Stringgoal) { |
3 | 6 | if (s.length() !=goal.length()) { |
4 | 7 | returnfalse; |
5 | 8 | } |
6 | | -intmismatchIdx = -1; |
| 9 | +intswapIdx = -1; |
7 | 10 | Set<Character>set =newHashSet<>(); |
8 | 11 | for (inti =0;i <s.length();i++) { |
| 12 | +if (s.charAt(i) ==goal.charAt(i)) { |
| 13 | +set.add(s.charAt(i)); |
| 14 | +continue; |
| 15 | + } |
9 | 16 | if (s.charAt(i) !=goal.charAt(i)) { |
10 | | -if (mismatchIdx ==Integer.MAX_VALUE) { |
| 17 | +if (swapIdx ==SWAP_DONE_IDX) { |
11 | 18 | returnfalse; |
12 | 19 | } |
13 | | -if (mismatchIdx == -1) { |
14 | | -mismatchIdx =i; |
15 | | - }else { |
16 | | -if (!(s.charAt(mismatchIdx) ==goal.charAt(i) &&s.charAt(i) ==goal.charAt(mismatchIdx))) { |
17 | | -returnfalse; |
18 | | - } |
19 | | -mismatchIdx =Integer.MAX_VALUE; |
| 20 | +if (swapIdx == -1) { |
| 21 | +swapIdx =i; |
| 22 | +continue; |
| 23 | + } |
| 24 | +if (!(s.charAt(swapIdx) ==goal.charAt(i) &&s.charAt(i) ==goal.charAt(swapIdx))) { |
| 25 | +returnfalse; |
20 | 26 | } |
21 | | - }else { |
22 | | -set.add(s.charAt(i)); |
| 27 | +swapIdx =SWAP_DONE_IDX; |
23 | 28 | } |
24 | 29 | } |
25 | | -if (mismatchIdx != -1 &&mismatchIdx !=Integer.MAX_VALUE) { |
26 | | -returnfalse; |
27 | | - } |
28 | | -if (mismatchIdx == -1) { |
29 | | -returns.length() >set.size(); |
30 | | - } |
31 | | -returntrue; |
| 30 | +return (swapIdx != -1 &&swapIdx ==SWAP_DONE_IDX) || (swapIdx == -1 &&set.size() <s.length()); |
32 | 31 | } |
33 | 32 | } |