leetcode第205场周赛

T1 替换所有的问号

• 题目链接

class Solution {
public:
    string modifyString(string s) {
        s = '#' + s + '#'; // 首尾添加哨兵
        int n = s.size(), t = 0;
        // 判断原s中的字符是否为?
        for (int i = 1; i < n - 1; i ++) {
        	// 当前字符为?时进行替换
            if (s[i] == '?') {
            	// 从a开始枚举
                char c = 'a';
                // 若c与前后任一字符相同，枚举下一个
                while (c == s[i - 1] || c == s[i + 1])
                	c++;
                // 替换
                s[i] = c;
            }
        }
        // 返回时去除前后哨兵
        s = s.substr(1, n - 2);
        return s;
    }
};

T2 数的平方等于两数乘积的方法数

• 题目链接

class Solution {
public:
    int numTriplets(vector<int>& nums1, vector<int>& nums2) {
        int res = 0;
        unordered_map<long long, int> hash;
        for (int i = 0; i < nums1.size(); i ++) {
            long long t = (long long)nums1[i] * nums1[i];
            // 对于每个nums1[i],hash要清零
            hash.clear();
            // 对于每个nums1[i]^2, 在nums2中寻找一对组合(j,k)使得nums1[i]^2=nums2[j]*nums2[k]
            for (int j = 0; j < nums2.size(); j ++) {
            	// nums1[i]^2能被nums[j]整除，则判断有没有nums1[i]^2/nums2[j]存在
                if (t % nums2[j] == 0) res += hash[t / nums2[j]], hash[nums2[j]] ++;
            }
        }
        for (int i = 0; i < nums2.size(); i ++) {
            long long t = (long long)nums2[i] * nums2[i];
            hash.clear();
            for (int j = 0; j < nums1.size(); j ++) {
                if (t % nums1[j] == 0) res += hash[t / nums1[j]], hash[nums1[j]] ++;
            }
        }
        return res;
    }
};

T3 避免重复字母的最小删除成本

• 题目链接

class Solution {
public:
    int minCost(string s, vector<int>& cost) {
        s += '#'; // 添加哨兵
        int sum = 0, maxc = INT_MIN, res = 0;
        for (int i = 0; i < s.length() - 1; i ++) {
        	// 计算当前重复字母删除成本之和，并记录单个字符最大删除成本
            sum += cost[i], maxc = max(maxc, cost[i]);
            // 若当前字符与下一个字符不同，则执行删除程序
            if (s[i] ！= s[i + 1]) {
            	// 总成本加上sum，若当前字母只有一个，则sum减去maxc后恰好为0
                res += sum - maxc;
                // 重置sum与maxc
                sum = 0;
                maxc = INT_MIN;
            }
        }
        return res;
    }
};

T4 保证图可完全遍历

• 题目链接

class Solution {
public:
	// 并查集
    int find(vector<int>& st, int x) {
        if (st[x] == x) return x;
        return st[x] = find(st, st[x]);
    }
    int maxNumEdgesToRemove(int n, vector<vector<int>>& e) {
        int m = e.size();
        int cnt = 0;
        vector<int> st1 = vector<int>(n + 1);
        // 并查集初始化
        for (int i = 1; i <= n; i ++) st1[i] = i;
        
        // 第一次克鲁斯卡尔算法，只针对第三种边，构建最小生成树
        for (int i = 0; i < m; i ++) {
            if (e[i][0] == 3) {
            	int a = e[i][1], b = e[i][2];
            	a = find(st1, a), b = find(st1, b);
            	if (a != b) {
                	st1[a] = b;
                	cnt ++ ;
            	}
            }
        }
        
        // st2拷贝于st1，分别用于构建alice和bob的最小生成树
        vector<int> st2(st1);
        int cnt1 = cnt, cnt2 = cnt;
        
        for (int i = 0; i < m; i ++) {
            int a = e[i][1], b = e[i][2];
            if (e[i][0] == 2) {
            	// bob的最小生成树，在第三种边的基础上添加bob专属边
            	a = find(st1, a), b = find(st1, b);
            	if (a != b) {
                	st1[a] = b;
                	cnt1 ++ ;
            	}
            }
            else if (e[i][0] == 1) {
            	// alice的最小生成树，在第三种边的基础上添加alice专属边
            	a = find(st2, a), b = find(st2, b);
            	if (a != b) {
                	st2[a] = b;
                	cnt2 ++ ;
                }
            }
        }
        
        // 若alice或bob的最小生成树不存在，返回-1
        if (cnt1 != n - 1 || cnt2 != n - 1) return -1;
        // 总边数减去alice和bob的最小生成树边数（就是n-1），加上他们的公共边数（被减了两次）
        return m - cnt1 - cnt2 + cnt;
    }
};