codeforce 954 div3 G2题

思路:

质因子分解可以顺着分解,也可以逆着分解

即找到每一个数字的倍数,再找到每一个数字的因数

cpp 复制代码
const int N = 5e5+10;
vector<int> ff[N];
vector<int> f[N];
vector<int> g[N];

void solve(){
    int n;
    cin>>n;
    vector<int> nums(n+1,0);
    ll ans = 0;
    for(int i = 1;i<=n;i++){
        cin>>nums[i];
    }
    for(int i = 1;i<=n;i++){
        int w = gcd(i,nums[i]);
        int x = i/w,y = nums[i]/w;
        //cout<<x<<" "<<y<<" eeeeee"<<endl;
        if(nums[i]%i == 0)  ans--;
        f[x].emplace_back(y);
        g[y].emplace_back(x);
    }
    vector<int>cnt(n+1,0);
    for(int i = 1;i<=n;i++){
        int m = f[i].size();
        if(m == 0)  continue;
        sort(f[i].begin(),f[i].end());
        for(int j = i;j<=n;j+=i){
            for(auto x : g[j]){
                cnt[x]++;
            }
        }
        for(int j = 0;j<m;j++){
            int k = 1;
            while(j<m-1 && f[i][j] == f[i][j+1]){
                j++;
                k++;
            }
            int y = f[i][j];
            ll sum = 0;
            for(auto son : ff[y]){
                sum += cnt[son];
            }
            sum *= k;
            ans += sum;
        }


        for(int j = i;j<=n;j+=i){
            for(auto x : g[j]){
                cnt[x]--;
            }
        }
    }
    for(int i = 1;i<=n;i++){
        f[i].clear();
        g[i].clear();
    }
    cout<<ans/2<<endl;
}
相关推荐
蒋星熠几秒前
Flutter跨平台工程实践与原理透视:从渲染引擎到高质产物
开发语言·python·算法·flutter·设计模式·性能优化·硬件工程
小欣加油27 分钟前
leetcode 面试题01.02判定是否互为字符重排
数据结构·c++·算法·leetcode·职场和发展
3Cloudream30 分钟前
LeetCode 003. 无重复字符的最长子串 - 滑动窗口与哈希表详解
算法·leetcode·字符串·双指针·滑动窗口·哈希表·中等
王璐WL38 分钟前
【c++】c++第一课:命名空间
数据结构·c++·算法
aramae44 分钟前
C++ -- 模板
开发语言·c++·笔记·其他
空白到白1 小时前
机器学习-聚类
人工智能·算法·机器学习·聚类
索迪迈科技1 小时前
java后端工程师进修ing(研一版 || day40)
java·开发语言·学习·算法
zzzsde2 小时前
【数据结构】队列
数据结构·算法
芒克芒克2 小时前
LeetCode 面试经典 150 题:删除有序数组中的重复项(双指针思想解法详解)
算法
青 .2 小时前
数据结构---二叉搜索树的实现
c语言·网络·数据结构·算法·链表