Longest Aritemetic Slices of Two sorted arrays

package _interview_question

/**
 * we have 2 arrays in sorted order, no duplicates.
we need find the max continuous length AP [arithmetic prog] in the first array by adding 0 or more elements from the second array
example,
A= [4,8,13]
B=[0,9,12]
ANSWER = 4, because: [0,4,8,12]
 * */
class Solution15 {
    /*
    * solution: merge two sorted array and find out longest arithmetic slices,
    * Time complexity:O(n^3), Space complexity:O(n)
    * */
    fun longestArithemeticSlices(nums1: IntArray, nums2: IntArray): Int {
        //merge two sorted array
        val n = nums1.size + nums2.size
        val array = IntArray(n)
        var i = 0
        var j = 0
        var index = 0
        while (i < nums1.size && j < nums2.size) {
            if (nums1[i] < nums2[j]) {
                array[index++] = nums1[i]
                i++
            } else {
                array[index++] = nums2[j]
                j++
            }
        }
        while (i < nums1.size) {
            array[index++] = nums1[i]
            i++
        }
        while (j < nums2.size) {
            array[index++] = nums2[j]
            j++
        }
        //find out longest arithmetic
        var max = 0
        for (i in 0 until n) {
            for (j in i + 1 until n) {
                val diff = array[j] - array[i]
                //for example: 1,2,3,4, A[j]=2, diff=1, so next should be 2+1
                var next = array[j] + diff
                //because has number 1,2, so current length is 2
                var currentMax = 2
                for (k in j + 1 until n) {
                    if (array[k] == next) {
                        currentMax++
                        next = array[k] + diff
                    }
                    max = Math.max(currentMax,max)
                }
            }
        }
        return max
    }
}

 

posted @ 2020-10-09 21:35  johnny_zhao  阅读(108)  评论(0编辑  收藏  举报