[题解]《算法零基础100讲》(第31讲) 多维枚举(一) - 入门（力扣392，240，2006，389，1431，1588，1534，771，1389，14，1925）

392. 判断子序列

双指针

用双指针扫一遍，找到相同给字符就给st+1

class Solution {
public:
    bool isSubsequence(string s, string t) {
        int st=0;
        for(int i=0;i<t.size();i++){
            if(s[st]==t[i]) st++;
        }
        return st==s.size();
    }
};

240. 搜索二维矩阵 II

二分

高效的查找算法，我们首先想到的当然就是二分。因为这道题的数组给的实在是太有规律了，所以我们发现显然有两段性，显然可以直接二分。稍微优化一下，因为每一行都是从小到大排列的，所以当发现某一行第一个元素都比目标值大的时候，如果还没有找到相等那么就可以直接返回false

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int st=0;
        int n=matrix.size(),m=matrix[0].size();
        while(st<n&&matrix[st][0]<=target) {
            int l=0,r=m-1;
            while(l<r) {
                int mid=l+r>>1;
                if(matrix[st][mid]>=target) r=mid;
                else l=mid+1;
            }
            if(matrix[st][l]==target) return true;
            st++;
        }
        return false;        
    }
};

2006. 差的绝对值为 K 的数对数目

线性循环

int countKDifference(int* nums, int numsSize, int k){
    int cnt=0;
    for(int i=0;i<numsSize;i++) {
        for(int j=i;j<numsSize;j++) {
            if(abs(nums[i]-nums[j])==k) cnt++;
        }
    }
    return cnt;
}

389. 找不同

枚举暴力查找

class Solution {
public:
    char findTheDifference(string s, string t) {
        unordered_map<char,int>mp1,mp2;
        for(auto c:s) mp1[c]++;
        for(auto c:t) mp2[c]++;
        for(auto c:mp2)
            if(mp1.find(c.first)==mp1.end()||mp1[c.first]!=c.second) return c.first;
        return ' ';
    }
};

1431. 拥有最多糖果的孩子

循环

先一次循环找到原来拥有糖果最多的孩子，然后依次比较每一个孩子和最大值的差值是否大于额外糖果

class Solution {
public:
    vector<bool> kidsWithCandies(vector<int>& candies, int extraCandies) {
        vector<bool>res;
        int maxn=0;
        for(auto t:candies) maxn=max(maxn,t);
        for(auto t:candies) res.push_back(maxn-t<=extraCandies);
        return res;
    }
};

1588. 所有奇数长度子数组的和

暴力循环

第一层循环用于枚举长度，第二层循环用于枚举起点，第三层循环用于枚举区间中每一个元素

class Solution {
public:
    int sumOddLengthSubarrays(vector<int>& arr) {
        int ans=0;
        for(int len=1;len<=arr.size();len+=2)
            for(int i=0;i<arr.size()-len+1;i++)
                for(int j=i;j<=i+len-1;j++)
                    ans+=arr[j];                   
        return ans;
    }
};

前缀和

我们观察发现最费劲的就是求和了。那么有没有简单一点的求和方式？前缀和就是很好的一种方式，他能在预处理前缀数组以后$O(1)$的时间复杂度直接算出一段连续区间的和。我们让$sum[1]=a[1],sum[2]=a[1]+a[2]=sum[1]+a[2],sum[3]=a[1]+a[2]+a[3]=sum[2]+a[3],...$这样在求$a[2]+a[3]$时就可以直接$sum[3]-sum[1]=a[1]+a[2]+a[3]-(a[1])=a[2]+a[3]$

class Solution {
public:
    int sumOddLengthSubarrays(vector<int>& arr) {
        int ans=0;
        int sum[110]={0};
        for(int i=1;i<=arr.size();i++) sum[i]=sum[i-1]+arr[i-1];
        for(int len=1;len<=arr.size();len+=2)
            for(int i=1;i<=arr.size()-len+1;i++)
                ans+=sum[i+len-1]-sum[i-1];
        return ans;
    }
};

1534. 统计好三元组

枚举

class Solution {
public:
    int countGoodTriplets(vector<int>& arr, int a, int b, int c) {
        int ans=0;
        for(int i=0;i<arr.size();i++)
            for(int j=i+1;j<arr.size();j++)
                for(int k=j+1;k<arr.size();k++)
                    if(abs(arr[i]-arr[j])<=a&&abs(arr[j]-arr[k])<=b&&abs(arr[i]-arr[k])<=c) ans++;
        return ans;
    }
};

771. 宝石与石头

枚举统计出现次数

class Solution {
public:
    int numJewelsInStones(string jewels, string stones) {
        int ans=0;
        int hash[256]={0};
        for(int i=0;i<jewels.size();i++) hash[jewels[i]]++;
        for(int i=0;i<stones.size();i++){
            if(hash[stones[i]]) ans++;
        }
        return ans;
    }
};

1389. 按既定顺序创建目标数组

枚举

class Solution {
public:
    vector<int> createTargetArray(vector<int>& nums, vector<int>& index) {
        vector<int>res;
        for(int i=0;i<index.size();i++)
            res.insert(res.begin()+index[i],nums[i]);
        return res;
    }
};

14. 最长公共前缀

枚举

依次枚举每一个字符串寻找他和之前串的最长公共子序列

class Solution {
public:
    string sub(string a,string b){
        string ch;
        for(int i=0;i<a.size()&&i<b.size();i++)
            if(a[i]==b[i]) ch+=a[i];
            else return ch;
        return ch;
    }
    string longestCommonPrefix(vector<string>& strs) {
        string ch=strs[0];
        for(int i=1;i<strs.size();i++){
            ch=sub(ch,strs[i]);
            if(ch=="") break;
        }
        return ch;
    }
};

1925. 统计平方和三元组的数目

枚举

class Solution {
public:
    int countTriples(int n) {
        int ans=0;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                for(int k=1;k<=n;k++)
                    if(i*i+j*j==k*k) ans++;
        return ans;
    }
};

枚举优化

$a^2+b^2=c^2$就等价于$b^2=c^2-a^2$所以就可以直接计算出两数平方差判断是不是一个平方数

class Solution {
public:
    int countTriples(int n) {
        int ans=0;
        for(int i=1;i<=n;i++)
            for(int j=i+1;j<=n;j++) {
                int v=j*j-i*i;
                int m=sqrt(v);
                if(m*m==v) ans++;
            }
        return ans;
    } 
};