1291 轻拍牛头（约数个数--反过来求解）

yuzhang_zy

已于 2022-03-05 09:36:59 修改

阅读量496

点赞数

分类专栏： acwing-提高文章标签：算法

于 2021-12-13 21:33:12 首次发布

本文链接：https://blog.csdn.net/qq_39445165/article/details/121914875

版权

约数计数优化算法计算机科学数学问题动态规划

关键词由CSDN通过智能技术生成

acwing-提高专栏收录该内容

456 篇文章 15 订阅

订阅专栏

1. 问题描述：

今天是贝茜的生日，为了庆祝自己的生日，贝茜邀你来玩一个游戏．贝茜让 N 头奶牛（编号 1 到 N）坐成一个圈。除了 1 号与 N 号奶牛外，i 号奶牛与 i−1 号和 i+1 号奶牛相邻，N 号奶牛与 1 号奶牛相邻。农夫约翰用很多纸条装满了一个桶，每一张纸条中包含一个 1 到 1000000 之间的数字。接着每一头奶牛 i 从桶中取出一张纸条，纸条上的数字用 Ai 表示。所有奶牛都选取完毕后，每头奶牛轮流走上一圈，当走到一头奶牛身旁时，如果自己手中的数字能够被该奶牛手中的数字整除，则拍打该牛的头。牛们希望你帮助他们确定，每一头奶牛需要拍打的牛的数量。即共有 N 个整数 A1，A2，…，AN，对于每一个数 Ai，求其他的数中有多少个是它的约数。

输入格式

第一行包含整数 N。接下来 N 行，每行包含一个整数 Ai。

输出格式

共 N 行，第 i 行的数字为第 i 头牛需要拍打的牛的数量。

数据范围

1 ≤ N ≤ 10 ^ 5，
1 ≤ Ai ≤ 10 ^ 6

输入样例：

5
2
1
2
3
4

输出样例：

2
0
2
1
3
来源：https://www.acwing.com/problem/content/description/1293/

2. 思路分析：

① 有关约数个数的一些性质：对于任何一个数字N都一定可以表示成下面的形式，其中p0，p1，p2...pk都是质因子：

N的所有质数个数为：

② 1~N的约数个数之和为：N / 1 + N / 2 + ... N / N = O(logN）

对于这道题目来说如果直接使用暴力枚举的话感觉不可取，因为求解每一个数的约数的时间复杂度为O(√n)，很难优化但是反过来求解比较简单，也即求解质数乘起来比较容易，所以我们可以统计A中数字出现的次数，将对应数字的出现的次数累加其所有不超过n的倍数上，这样就很容易处理了。

3. 代码如下：

class Solution:
    # 反过来求解, 累加到对应的倍数
    def process(self):
        N = 10 ** 6 + 10
        n = int(input())
        a = list()
        for i in range(n):
            a.append(int(input()))
        cnt, s = [0] * N, [0] * N
        for i in range(n):
            cnt[a[i]] += 1
        for i in range(1, N):
            for j in range(i, N, i):
                s[j] += cnt[i]
        for i in range(n):
            # 注意不包括自己
            print(s[a[i]] - 1)


if __name__ == '__main__':
    Solution().process()