力扣 第493场周赛(A~D)

A:3870. 统计范围内的逗号

三位一个逗号，问1~n总共有多少个逗号。

数据范围较小，可直接模拟。

class Solution {
public:
    int countCommas(int n) {
        int ans=0;
        for(int i=1;i<=n;i++){
            int t=i;
            while(t/1000>0){
                ans++;
                t/=1000;
            }
        }
        return ans;
    }
};

B:3871. 统计范围内的逗号 II

A的升级版，升级了数据范围，不能直接模拟了。

赛时写了个很复杂的，后面听了题解，横看成岭侧成峰啊，以乘每1000为一个阶段，当前阶段内每一个数的逗号数为c，那么下一个阶段的逗号数为c+1。

typedef long long LL;
class Solution {
public:
    long long countCommas(long long n) {
        LL ans=0;
        LL c=0,m=1;
        while(m*1000<=n){
            ans+=(1000*m-1-m+1)*c;
            m*=1000;
            c++;
        }
        ans+=(n-m+1)*c;
        return ans;
    }
};

C:3872. 替换最多一个元素后的最长等差子数组
前后缀分解，注意除了pre[i-1]+1+suf[i+1]和pre[i-1]+1两种情况之外，还需要考虑两边d不相同，但是右边的第一个元素能够匹配左边的d的情况，反过来同理。

class Solution {
public:
    int longestArithmetic(vector<int>& nums) {
        int n=nums.size();
        auto calc=[&](vector<int> t){
            vector<int> res(n,0);
            res[0]=1;
            res[1]=2;
            for(int i=2;i<n;i++){
                if(t[i]+t[i-2]==t[i-1]*2)
                    res[i]=res[i-1]+1;
                else
                    res[i]=2;
            }
            return res;
        };
        vector<int> pre=calc(nums);
        reverse(nums.begin(),nums.end());
        vector<int> suf=calc(nums);
        reverse(suf.begin(),suf.end());
        int ans=(*max_element(pre.begin(),pre.end()))+1;
        if(ans>=n)
            return n;
        reverse(nums.begin(),nums.end());
        for(int i=1;i<n-1;i++){
            int d2=nums[i+1]-nums[i-1];
            if(d2%2)
                continue;
            bool ok_left=i>1 && (nums[i-1]-nums[i-2]==d2/2);
            bool ok_right=i+2<n && (nums[i+2]-nums[i+1]==d2/2);
            if(ok_left && ok_right)
                ans=max(ans,pre[i-1]+1+suf[i+1]);
            else if(ok_left)
                ans=max(ans,pre[i-1]+2);
            else if(ok_right)
                ans=max(ans,suf[i+1]+2);
        }
        return ans;
    }
};

 

D:3873. 添加一个点后可激活的最大点数
一眼并查集，将具有相同x或相同y的点都连成一个集合，新加一个点可以任选两个集合激活。那么就选最大的两个，在并查集的同时计算一下cnt即可。

class Solution {
public:
    int maxActivated(vector<vector<int>>& pts) {
        int n=pts.size();
        vector<int> f(n);
        vector<int> cnt(n,1);
        for(int i=0;i<n;i++)
            f[i]=i;
        function<int(int)> find=[&](int x){
            if(f[x]!=x)
                f[x]=find(f[x]);
            return f[x];
        };
        auto merge=[&](int a,int b){
            int fa=find(a),fb=find(b);
            if(fa!=fb){
                f[fb]=fa;
                cnt[fa]+=cnt[fb];
            }
        };
        map<int,vector<int>> xg,yg;
        for(int i=0;i<n;i++){
            int x=pts[i][0],y=pts[i][1];
            xg[x].push_back(i);
            yg[y].push_back(i);
        }
        for(auto [k,v]:xg){
            int t=v[0];
            for(int i=1;i<v.size();i++){
                merge(t,v[i]);
            }
        }
        for(auto [k,v]:yg){
            int t=v[0];
            for(int i=1;i<v.size();i++){
                merge(t,v[i]);
            }
        }
        vector<int> s;
        vector<bool> vis(n,false);
        for(int i=0;i<n;i++){
            int t=find(i);
            if(!vis[t]){
                s.push_back(cnt[t]);
                vis[t]=true;
            }
        }
        sort(s.begin(),s.end(),[](int a,int b){return a>b;});
        if(s.size()==0)
            return 1;
        else if(s.size()==1)
            return s[0]+1;
        else
            return s[0]+s[1]+1;
    }
};