Leetcode 题组 19（set）

349. 两个数组的交集

给定两个数组，编写一个函数来计算它们的交集。

示例 1：

输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2]

示例 2：

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> m1(nums1.begin(), nums1.end()), m2(nums2.begin(), nums2.end());
        vector<int> ans;
        for (int it : m1){
            if (m2.count(it)) ans.push_back(it);
        }
        return ans;
    }
};

 242. 有效的字母异位词

给定两个字符串 s 和 t ，编写一个函数来判断 t 是否是 s 的字母异位词。

注意：若 s 和 t 中每个字符出现的次数都相同，则称 s 和 t 互为字母异位词。

示例 1:

输入: s = "anagram", t = "nagaram"
输出: true

class Solution {
public:
    bool isAnagram(string s, string t) {
        vector<int> vs(128, 0), vt(128, 0);
        if (s.length()!=t.length()) return false; // 如果是异位词，那么长度一定是相等的。
        for (int i = 0; i < s.length(); ++i){
            ++vs[s[i]];
            ++vt[t[i]]; 
        }
        return vs==vt ;
    }
};

 202. 快乐数

编写一个算法来判断一个数 n 是不是快乐数。

「快乐数」定义为：

对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
如果 可以变为  1，那么这个数就是快乐数。
如果 n 是快乐数就返回 true ；不是，则返回 false 。

示例 1：

输入：n = 19
输出：true
解释：
1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1

class Solution {
public:

    int nextN(int n){
        int ret = 0;
        while(n){
            int tmp = n % 10;
            ret+=tmp*tmp;
            n/=10;
        }
        return ret;
    }

    bool isHappy(int n) {
        unordered_set<int> s;
        while(n!=1){
            if (s.count(n)) return false;
            s.insert(n);
            n = nextN(n);
        }
        return true;
    }
};

454. 四数相加 II 

给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
 

示例 1：

输入：nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出：2
解释：
两个元组如下：
1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> m;
        int ans = 0;
        for (int a : nums1)
            for (int b : nums2)
                m[a+b]++;
        for (int c : nums3)
            for (int d : nums4)
                if (m.find(0-(c+d))!=m.end())
                    ans += m[0-(c+d)];
        return ans;
    }
};