将这种
for(int i=1;i<=n;i++){ ans+=(s0[s[i][0]].size()-1)*(s1[s[i][1]].size()-1); } }
改为
int len0=s0[s[i][0]].size(),len1=s1[s[i][1]].size(); ans+=(len0-1)*(len1-1);