【1790. 仅执行一次字符串交换能否使两个字符串相等】

一时失言乱红尘 2023-10-03 22:55 19阅读 0赞

package com.lfsun.main.demoleetcode.easy;
    
    import org.junit.Test;
    
    /**
     * 给你长度相等的两个字符串 s1 和 s2 。一次 字符串交换 操作的步骤如下：选出某个字符串中的两个下标（不必不同），并交换这两个下标所对应的字符。
     *
     * 如果对 其中一个字符串 执行 最多一次字符串交换 就可以使两个字符串相等，返回 true ；否则，返回 false 。
     *
     * 示例 1：
     *
     * 输入：s1 = "bank", s2 = "kanb"
     * 输出：true
     * 解释：例如，交换 s2 中的第一个和最后一个字符可以得到 "bank"
     * 示例 2：
     *
     * 输入：s1 = "attack", s2 = "defend"
     * 输出：false
     * 解释：一次字符串交换无法使两个字符串相等
     * 示例 3：
     *
     * 输入：s1 = "kelb", s2 = "kelb"
     * 输出：true
     * 解释：两个字符串已经相等，所以不需要进行字符串交换
     * 示例 4：
     *
     * 输入：s1 = "abcd", s2 = "dcba"
     * 输出：false
     * 提示：
     *
     * 1 <= s1.length, s2.length <= 100
     * s1.length == s2.length
     * s1 和 s2 仅由小写英文字母组成
     * @author: lfsun
     * @time: 17:44
     * @date: 2022/11/19
     */
    public class AreAlmostEqual {
        
        public boolean areAlmostEqual(String s1, String s2) {
        
            if (s1.length() != s2.length()) return false;
            char c1 = 0, c2 = 0;
            int notEqualSize = 0;
            for (int i = 0; i < s1.length(); i++) {
        
                char cs1 = s1.charAt(i);
                char cs2 = s2.charAt(i);
                if (cs1 != cs2) {
        
                    if (notEqualSize == -1) return false;
                    notEqualSize++;
                    if (notEqualSize > 2) return false;
                    if (notEqualSize == 1) {
        
                        c1 = cs1;
                        c2 = cs2;
                        continue;
                    }
                    if (cs1 == c2 && cs2 == c1) notEqualSize = -1;
    
                }
            }
            return notEqualSize == 0 || notEqualSize == -1;
        }
    
        public boolean areAlmostEqual2(String s1, String s2) {
        
            if (s1.length() != s2.length()) return false;
            int count = 0, m = -1, n = -1;
            for (int i = 0, j = 0; i < s1.length(); i++, j++) {
        
                if (s1.charAt(i) != s2.charAt(j)) {
        
                    count++;
                    if (m == -1) m = i;
                    if (n == -1 && m != i) n = i;
                }
            }
            if (count == 0) return true;
            if (count == 2 && s1.charAt(m) == s2.charAt(n) && s2.charAt(m) == s1.charAt(n))
                return true;
            return false;
        }
    
        @Test
        public void test() {
        
            System.out.println(areAlmostEqual("aa", "ac"));
            System.out.println(areAlmostEqual("siyolsdcjthwsiplccjbuceoxmpjgrauocx", "siyolsdcjthwsiplccpbuceoxmjjgrauocx"));
            System.out.println(areAlmostEqual("bank", "kanb"));
        }
    }