《算法竞赛·快冲300题》每日一题：“Pow Set”

《算法竞赛·快冲300题》将于2024年出版，是《算法竞赛》的辅助练习册。
所有题目放在自建的OJ New Online Judge。
用C/C++、Java、Python三种语言给出代码，以中低档题为主，适合入门、进阶。

“ Pow Set” ，链接： http://oj.ecustacm.cn/problem.php?id=1797

题目描述

【题目描述】 给定n,m，考虑以下集合：S = {a^b|2≤a≤n, 2≤b≤m}，其中a^b表示a的b次方，求集合S去重后有多少个元素。
【输入格式】 两个正整数n和m。（2≤n,m≤500）。
【输出格式】 输出一个数字表示答案。
【输入样例】

5 5

【输出样例】

15

---

题解

  本题很直白，首先算幂$a^b$，然后判重。集合S内的元素最多有n×m个，由于n和m都不大，可以直接算。
【重点】 大数的计算。

C++代码

  如果用C++编码，直接计算$a^b$需要用高精度，而高精度的编码比较麻烦，而且a^b的值太大，判重也比较困难。
  如果对$a^b$取模，不需要用精度编码，但是不同的数取模后可能相同，导致统计结果出错。为保险起见，不要用这个方法。
  下面的代码不直接算$a^b$，而是把它分解为质因数的幂，例如$24^3=2^9\times 3^3$。这样计算准确且不会有取模的问题。用set对分解后的质因数和幂次判重。

#include<bits/stdc++.h>
using namespace std;
typedef pair <int, int> Pair;
vector<Pair> factor(int x){                           //对x进行因式分解
    vector<Pair> ans;
    for(int i = 2; i * i <= x; i++)if(x % i == 0) {   //i是x的因子
        int num = 0;
        while(x % i == 0)++num, x /= i;               //统计i的幂
        ans.push_back(make_pair(i, num));             //存入ans
    }
    if(x != 1)                                        //最后可能有一个大于根号x的素因子
        ans.push_back(make_pair(x, 1));
    return ans;
}
vector<Pair>all_factor[510];                          //all_factor[i]表示i的唯一分解
set<vector<Pair>>S;                                   //存储集合
void insert_pow(int a, int b){                        //把a^b以唯一分解的形式塞入集合S
    vector<Pair>ans;
    for(auto x : all_factor[a])                       //遍历a的唯一分解
        ans.push_back(make_pair(x.first, x.second * b));
    S.insert(ans);
}
int main(){
    for(int i = 2; i <= 500; i++)           //预处理出2到500每个数字的唯一分解表示
        all_factor[i] = factor(i);
    int n, m;    cin >> n >> m;
    for(int a = 2; a <= n; a++)
        for(int b = 2; b <= m; b++)
            insert_pow(a, b);
    cout<<S.size();
    return 0;
}

Java代码

  Java可以直接算大数，然后用set判重，代码非常简单。

import java.math.BigInteger;
import java.util.HashSet;
import java.util.Scanner; 
public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n=scanner.nextInt();
        int m=scanner.nextInt();
        HashSet<BigInteger> set = new HashSet<BigInteger>();         
        for(int i=2;i<=n;i++)
            for(int j=2;j<=m;j++)
                set.add(BigInteger.valueOf(i).pow(j));          
        System.out.println(set.size());         
    }
}

Python代码

  Python可以直接算大数，然后用set或者字典判重，代码非常简单。

n,m = map(int,input().split())
s = set()
for i in range(2,n+1):
    for j in range(2,m+1):
        s.add(i**j)
print(len(s))