【刷题打卡】1035. 不相交的线

148 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

一、题目描述:

1035. 不相交的线 - 力扣(LeetCode) (leetcode-cn.com)

在两条独立的水平线上按给定的顺序写下 nums1nums2 中的整数。

现在,可以绘制一些连接两个数字 nums1[i] 和 nums2[j] 的直线,这些直线需要同时满足满足:

  •  nums1[i] == nums2[j]
  • 且绘制的直线不与任何其他连线(非水平线)相交。

请注意,连线即使在端点也不能相交:每个数字只能属于一条连线。

以这种方法绘制线条,并返回可以绘制的最大连线数。

示例 1:

image.png

输入:nums1 = [1,4,2], nums2 = [1,2,4]
输出:2
解释:可以画出两条不交叉的线,如上图所示。 
但无法画出第三条不相交的直线,因为从 nums1[1]=4 到 nums2[2]=4 的直线将与从 nums1[2]=2 到 nums2[1]=2 的直线相交。

示例 2:

输入:nums1 = [2,5,1,2,5], nums2 = [10,5,2,1,5,2]
输出:3

示例 3:

输入:nums1 = [1,3,7,1,7,5], nums2 = [1,9,2,5,1]
输出:2

 

提示:

  • 1 <= nums1.length, nums2.length <= 500
  • 1 <= nums1[i], nums2[j] <= 2000  

二、思路分析:

可以用二维动态规划来做这道题,同时可以看到,二位状态矩阵其实只用到了当前列和上一列,因此可以考虑只用一维数组来进一步优化

dp[i][j]转移自dp[i-1][j],dp[i][j-1],dp[i-1][j-1]

我们可以省略掉i维,只保存j维的值,那么dp(i)[j]转移自dp(i-1)[j],dp(i)[j-1],dp(i-1)[j-1]

注意括号内的i和i-1代表第i次循环和第i-1次循环。因为dp(i-1)[j-1]会在第i次循环的时候被覆盖掉,所以我们额外保存一个dp(i-1)[j-1]

三、AC 代码:

class Solution {
public:
    
    int maxUncrossedLines(vector<int>& nums1, vector<int>& nums2) {
        int dp[nums2.size()+2];
        memset(dp,0,sizeof(dp));
        for(int i = 1; i <= nums1.size(); ++i)
        {
            int last = dp[0], temp;
            for(int j = 1; j <= nums2.size(); ++j)
            {
                temp = dp[j];
                dp[j] = max(dp[j], dp[j-1]);
                if(nums1[i-1] == nums2[j-1]) dp[j] = max(dp[j], last+1);
                else dp[j] = max(dp[j], last);
                last = temp;
            }
        }
        return dp[nums2.size()];
    }