力扣 1128. 等价多米诺骨牌对的数量思维

最新推荐文章于 2022-11-12 10:38:41 发布

csu_xiji

最新推荐文章于 2022-11-12 10:38:41 发布

阅读量132

点赞数

分类专栏：力扣思维

本文链接：https://blog.csdn.net/xiji333/article/details/113182612

版权

力扣同时被 2 个专栏收录

282 篇文章 2 订阅

订阅专栏

思维

181 篇文章 1 订阅

订阅专栏

https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs/
在这里插入图片描述
思路一：由于二元对的取值范围很小，不妨用 $cnt_{ij}$ 表示输入序列中二元对 $(i, j)$ 的数量，我们在遍历输入序列的同时，计算这个 $c n t$ 数组，那么当遍历到第 $k$ 个元素时， $c n t$ 已经统计了 $[0, k - 1]$ 的数据，此时对于第 $k$ 个元素，我们可以计算出与它等价的骨牌数量： $cnt[k_0][k_1]+cnt[k_1][k_0]$ 。注意当 $k_0=k_1$ 时，我们只需要计算一次：

class Solution {
public:
    int numEquivDominoPairs(vector<vector<int>>& dominoes) {
        vector<vector<int>> cnt(10,vector<int>(10));
        int ans=0;
        for(vector<int>&ele : dominoes)
        {
            ans+=cnt[ele[0]][ele[1]];
            if(ele[0]!=ele[1])
                ans+=cnt[ele[1]][ele[0]];
            ++cnt[ele[0]][ele[1]];
        }
        return ans;
    }
};

思路二：可以只使用一维数组，且严格限定二元对的 $i > = j$ ，其它思路同上：

class Solution {
public:
    int numEquivDominoPairs(vector<vector<int>>& dominoes) {
        int arr[100]={0};
        int ans=0;
        for(vector<int>&ele : dominoes)
        {
            int val=ele[0]>=ele[1]?ele[0]*10+ele[1]:ele[1]*10+ele[0];
            ans+=arr[val];
            ++arr[val];
        }
        return ans;
    }
};