【412】【统计近似相称数对 II】

差130个样例，等佬解

2. class Solution:
3.     def ifqual(self,str1,str2):
4.         return int(str1)==int(str2)
5.     def change(self,str1,str2):
6.         str1 = list(str1)
7.         n=len(str1)
8.         t=0
9.         for i in range(n):
10.             for j in range(i+1,n):
11.                 str1[i],str1[j]=str1[j],str1[i]
12.                 t+=1
13.                 if self.ifqual("".join(str1),str2):
14.                     return True
15.                 if t==2:
16.                     str1[i],str1[j]=str1[j],str1[i]   
17.         return False
18.          
19.     def countPairs(self, nums: List[int]) -> int:
20.         n=len(nums)
21.         ret=0
22.         for i in range(n):
23.             for j in range(i+1,n):
24.                 if nums[i]==nums[j]:
25.                     ret+=1
26.                 else:
27.                    num1=str(nums[i])
28.                    num2=str(nums[j])
29.                    if self.change(num1,num2) or self.change(num2,num1):
30.                        ret+=1
31.         return ret

复制代码

 

大佬题解

2. class Solution:
3.     def countPairs(self, nums: List[int]) -> int:
4.         nums.sort()
5.         ret=0
6.         dict=defaultdict(int)
7.         for num in nums:
8.             #不交换
9.             st={num}
10.             #num的长度
11.             s=list(str(num))
12.             m=len(s)
13.             #数位交换
14.             for i in range(m):
15.                 for j in range(i+1,m):
16.                     s[i],s[j]=s[j],s[i]
17.                     st.add(int(''.join(s)))
18.                     for p in range(i+1,m):
19.                       for q in range(p+1,m):
20.                         s[p],s[q]=s[q],s[p]
21.                         st.add(int(''.join(s)))
22.                         #交换2次
23.                         s[p],s[q]=s[q],s[p]
24.                     #换回来
25.                     s[i],s[j]=s[j],s[i]
26.             ret+=sum(dict[v] for v in st)
27.             dict[num]+=1
28.         return ret         

复制代码

思绪确实好用，2题都能解。
ps:
力扣刷完还能atcoder>codeforces.