记一次字节面试的两道代码题
1. [数组中的逆序对][https://leetcode-cn.com/problems/shu-zu-zhong-de-ni-xu-dui-lcof/]
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。对于结果取模1000000007。题目保证输入的数组中没有的相同的数字。
输入: [7,5,6,4]
输出: 5
开始的时候就直接用双层循环遍历的,毫无疑问超时了。。。所以需要降低时间复杂度。
使用归并排序,不断进行二分直至子数组。长度为1,然后进行合并。在合并的过程中,左边子数组的当前元素值大于右边子数组的当前值时,则是一个逆序对。代码如下:
class Solution{
int[] tmp;
public int reversePairs(int[] nums){
tmp = new int[nums.length];
return mergeSort(0, nums, nums.length - 1);
}
public int mergeSort(int l, int[] nums, int r){
if(l>=r){
return 0;
}
int m = (l + r)/2;
int res = mergeSort(l, nums, m) + mergeSort(m+1, nums, r);
int i = l, j = m+1;
for(int k=l; k<=r; k++){
tmp[k] = nums[k];
}
for(int k=l; k<=r; k++){
if(i == m+1){
nums[k] = tmp[j++];
}else if(j == r+1 || tmp[i] <= tmp[j]){
nums[k] = tmp[i++];
}else{
nums[k] = tmp[j++];
res = res + m - i + 1;
}
}
return res;
}
}
2. 最小编辑代价
给定两个字符串str1和str2,再给定三个整数ic,dc和rc,分别代表插入、删除和替换一个字符的代价,请输出将str1编辑成str2的最小代价。
与最小编辑距离类似,只是在其基础上对各个操作加上了代价,需要进行分类,仍然还是动态规划。代码如下:
import java.util.*;
public class Solution{
public int minEditCost(String str1, String str2, int ic, int dc, int rc){
int[][] dp = new int[str1.length()+1][str2.length()+1];
for(int i=0; i<dp[0].length; i++){
dp[0][i] = i*ic;
}
for(int i=0; i<dp.length; i++){
dp[i][0] = i*dc;
}
for(int i=1; i<dp.length; i++){
for(int j=1; j<dp[0].length; j++){
if(str1.charAt(i-1) == str2.charAt(j-1)){
dp[i][j] = dp[i-1][j-1];
}else{
dp[i][j] = Math.min(dp[i-1][j]+dc, Math.min(dp[i-1][j-1]+rc, dp[i][j-1]+ic));
}
}
}
return dp[str1.length()][str2.length()];
}
}
发出来希望对大家能有点帮助吧。