【面试算法题总结16】数学相关算法

数学相关算法：

 

例题1：数值的整数次方

即快速幂
n=-2147483648这个用例决定了要用long变量b。
下面(b % 2) == 1可以变成(b & 1) == 1；b /=2;可以变成b >>= 1;

class Solution {
    public double myPow(double x, int n) {
        if(x == 0) return 0;
        long b = n;
        double res = 1.0;
        if(b < 0) {
            x = 1 / x;
            b = -b;
        }
        while(b > 0){
            if((b % 2) == 1){
                res *= x;
            }
            x *= x;
            b /=2;
        }
        return res;
    }
}

 

例题2：已知xy和x+y，求xⁿyⁿ

每个运算都%mod，而且还要防止一个大数-小数，但只有大数被%mod了，还要再+mod，最后再%mod。

import java.util.*;

public class Main{
    public static void main(String[] args) {
        Scanner in =new Scanner(System.in);
        int T=in.nextInt();
        for(int i=0;i<T;++i){
            int A=in.nextInt();
            int B=in.nextInt();
            int n=in.nextInt();

            //逻辑开始
            long mod=(long)Math.pow(10,9)+7;
            long temp1=2;
            long temp2=A;
            for(int k=2;k<=n;++k){
                long temp=((A*temp2%mod)-(B*temp1%mod)+mod)%mod;
                temp1=temp2;
                temp2=temp;
            }
            System.out.println(temp2);
        }
    }
}

 

例题3：确定比例，求最大乘积

import java.util.*;

public class Main{

    public static void main(String[] args) {
        Scanner in =new Scanner(System.in);
        long A=in.nextLong();
        long B=in.nextLong();
        long a=in.nextLong();
        long b=in.nextLong();

        //逻辑开始
        //找到a和b的最大公约数(让a和b同比变小)
        long c=f(a,b);
        long minX=a/c;
        long minY=b/c;
        
        long resultX=0,resultY=0;
        for(int k=1;k*minX<=A&&k*minY<=B;++k){
            long x=k*minX;
            long y=k*minY;
            resultX=x;
            resultY=y;
        }
        System.out.println(resultX+" "+resultY);
    }

    //找到a和b的最大公约数
    static long f(long a,long b){
        long c=0;
        while(b!=0){
            c=a%b;
            a=b;
            b=c;
        }
        return a;
    }
}

 

例题4：修水渠

import java.util.*;

public class T12{
    public static void main(String[] args) {
        Scanner in =new Scanner(System.in);
        int n=in.nextInt();
        int[] house=new int[n];
        for(int i=0;i<n;++i){
            house[i]=in.nextInt();
            in.nextInt();
        }

        //逻辑开始
        Arrays.sort(house);
        int midIndex=n/2;
        long result=0;
        for(int i=0;i<n;++i){
            result+=Math.abs(house[midIndex]-house[i]);
        }
        System.out.println(result);
    }
}

 

例题5：安排一下座位，达到邻座之间的差异之和最大

import java.util.*;

public class T13{
    public static void main(String[] args) {
        Scanner in =new Scanner(System.in);
        int n=in.nextInt();
        int[] nums=new int[n];
        for(int i=0;i<n;++i){
            nums[i]=in.nextInt();
        }

        //逻辑开始
        Arrays.sort(nums);

        // 声明一个输出空间
        long[] resultArr = new long[n];
        long reslutSum = 0;

        int left = 0;
        int right = n - 1;
        for (int i = 0; i < n; i++){
            if(i%2==0){
                resultArr[i] = nums[left];
                left++;
            }else {
                resultArr[i] = nums[right];
                right--;
            }

            if(i>0) reslutSum = reslutSum + Math.abs(resultArr[i]-resultArr[i-1]);
        }
        //计算最后一位与第一位的差值
        reslutSum = reslutSum + Math.abs(resultArr[n-1]-resultArr[0]);

        //输出结果
        System.out.println(reslutSum);
        for (long result : resultArr) {
            System.out.print(result + " ");
        }
    }
}

 

例题6：正n边形的等腰锐角三角形数量

画了一个正n边形，他想知道多边形中等腰锐角三角形的数量。（三角形的顶点要在多边形的顶点上）

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner in =new Scanner(System.in);
        int n=in.nextInt();
        //计算没有减去钝角前每个角能构成的等腰三角形个数p
        long p=(n-1)/2;
        //p减去其中的钝角个数
        p=(p+1)/2;
        //计算减去钝角后的结果sum
        long sum=n*p;
        //sum减去直角的个数
        if(n%4==0)sum-=n;
        //sum减去重复的锐角的个数
        if(n%3==0)sum-=n/3*2;
        System.out.println(sum);
    }
}

 

例题7：表示数值的字符串

class Solution {
    int index=0;
    public boolean isNumber(String s) {
        if(s.length()==0){
            return false;
        }
        s=s.trim();
        s=s+"|";
        boolean result=isInt(s);
        //处理有小数点的情况
        if(s.charAt(index)=='.'){
            ++index;
            result|=isPositive(s);      //小数点后可以什么都没有
        }
        //处理有e和E的情况
        if(s.charAt(index)=='e'||s.charAt(index)=='E'){
            ++index;
            result&=isInt(s);       //e或E后必须有个整数
        }
        return result&&s.charAt(index)=='|';
    }
    //判断是否是整数
    public boolean isInt(String s){
        if(s.charAt(index)=='-'||s.charAt(index)=='+'){
            ++index;
        }
        return isPositive(s);
    }
    //判断是否是正整数
    public boolean isPositive(String s){
        int start=index;
        while(s.charAt(index)>='0'&&s.charAt(index)<='9'){
            ++index;
        }
        return start!=index;
    }
}

 

例题8：数组中出现次数超过一半的数字

class Solution {
    public int majorityElement(int[] nums) {
        int time=0;
        int result=0;
        for(int i=0;i<nums.length;++i){
            if(time==0){
                result=nums[i];
                time=1;
                continue;
            }
            if(result==nums[i]){
                ++time;
            }else{
                --time;
            }
        }
        return result;
    }
}

 

例题9：数据流中的中位数

class MedianFinder {
    Queue<Integer> A,B;
    /** initialize your data structure here. */
    public MedianFinder() {
        //小根堆
        A=new PriorityQueue<>();
        //大根堆
        B=new PriorityQueue<>((e1,e2)->e2-e1);
    }
    
    public void addNum(int num) {
        if(A.size()!=B.size()){
            A.offer(num);
            B.offer(A.poll());
        }else{
            B.offer(num);
            A.offer(B.poll());
        }
    }
    
    public double findMedian() {
        return A.size()!=B.size()?(double)A.peek():(double)(A.peek()+B.peek())/2;
    }
}

/**
 * Your MedianFinder object will be instantiated and called as such:
 * MedianFinder obj = new MedianFinder();
 * obj.addNum(num);
 * double param_2 = obj.findMedian();
 */

 

例题10：1～n 整数中 1 出现的次数

大佬题解

class Solution {
    public int countDigitOne(int n) {
        int digit=1,res=0;
        int high=n/10,cur=n%10,low=0;
        while(high!=0||cur!=0){
            if(cur==0){
                res+=high*digit;
            }else if(cur==1){
                res+=high*digit+low+1;
            }else{
                res+=(high+1)*digit;
            }
            low+=cur*digit;
            cur=high%10;
            high/=10;
            digit*=10;
        }
        return res;
    }
}

 

例题11：数字序列中某一位的数字

这里注意start和count必须为long，因为start和count超过int范围后，还要和n比较一次。

class Solution {
    public int findNthDigit(int n) {
        int digit=1;        //位数。如245的位数为3
        long start=1;        //每digit位数的起始数字（即：1, 10, 100⋯）。如digit为3时，start为100
        long count=9;        // 一位数、两位数、...、(digit-1)位数的数位数量count,count=9*start*digit。即digit为3时，count为9*100*3
        //第一步，算出n的所在的digit
        while(n>count){
            n-=count;
            start*=10;
            digit+=1;
            count=9*start*digit;
        }
        //算出n所在数字
        long num=start+(n-1)/digit;
        //确定所求数位在 numnum 的哪一数位
        return Long.toString(num).charAt((n-1)%digit)-'0';
    }
}

 

例题12：旋转图像

核心在于：matrix[j][n-i-1]=matrix[i][j];
通过拆分可以变成先水平翻转，再对角线翻转。

class Solution {
    public void rotate(int[][] matrix) {
        int n=matrix.length;
        //水平翻转
        for(int i=0;i<n/2;++i){
            for(int j=0;j<n;++j){
                int temp=matrix[i][j];
                matrix[i][j]=matrix[n-i-1][j];
                matrix[n-i-1][j]=temp;
            }
        }
        //对角线翻转
        for(int i=0;i<n;++i){
            for(int j=0;j<i;++j){
                int temp=matrix[i][j];
                matrix[i][j]=matrix[j][i];
                matrix[j][i]=temp;
            }
        }
    }
}

 

例题15：62. 不同路径

解法1：动态规划算法

class Solution {
    public int uniquePaths(int m, int n) {
        //dp[i][j]表示到达[i][j]位置的路径数
        int[][] dp=new int[m][n];

        for(int i=0;i<m;++i){
            for(int j=0;j<n;++j){
                if(i==0||j==0){ //边界条件
                    dp[i][j]=1;
                    continue;
                }
                dp[i][j]=dp[i-1][j]+dp[i][j-1];
            }
        }

        return dp[m-1][n-1];
    }
}

解法2：数学组合

一共要走m+n-2步，就是确定其中那m-1步向下选在哪些位置即可，即：$C_{m+n+2}^{m-1}$

class Solution {
    public int uniquePaths(int m, int n) {
        //求组合
        long result=1;
        for(int i=1;i<=m-1;++i){		//不可以--i，不然会出现1/2=0的问题
            result=result*(i+n-1)/i;
        }
        return (int)result;
    }
}

 

例题16：字典序的第K小数字

字典树思想

class Solution {
    public int findKthNumber(int n, int k) {
        int curr = 1;
        k--;
        while (k > 0) {
            int steps = getSteps(curr, n);
            if (steps <= k) {
                k -= steps;
                curr++;
            } else {
                curr = curr * 10;
                k--;
            }
        }
        return curr;
    }

    public int getSteps(int curr, long n) {
        int steps = 0;
        long first = curr;
        long last = curr;
        while (first <= n) {
            steps += Math.min(last, n) - first + 1;
            first = first * 10;
            last = last * 10 + 9;
        }
        return steps;
    }
}