acwing::acwing4319 命名空间参考

1. 合适数对

更多...

函数
int	main (istream &cin, ostream &cout)
	TEST (acwing4319, case1)

详细描述

1. 合适数对

函数说明

main()

int acwing::acwing4319::main	(	istream &	cin,
		ostream &	cout
	)

在文件 acwing.cpp 第 5898 行定义.

                                              {
            unsigned int n;
            unsigned int k;
            cin >> n >> k;
            vector<unsigned int> a(n);
            vector<vector<pair<unsigned int, unsigned int>>> factors(n);
            map<vector<pair<unsigned int, unsigned int>>, unsigned int> factor_status_count;
            for(int i = 0; i < n; i++) {
                cin >> a[i];
                for(unsigned int factor = 2; factor * factor <= a[i]; factor++) {
                    if(a[i] % factor != 0U) {
                        continue;
                    }
                    unsigned int count = 0;
                    while(a[i] % factor == 0) {
                        a[i] /= factor;
                        ++count;
                    }
                    count %= k;
                    if(count != 0) {
                        factors[i].emplace_back(factor, count);
                    }
                }
                if(a[i] != 1) {
                    factors[i].emplace_back(a[i], 1);
                }
            }
            unsigned long long ans = 0;
            for(unsigned int i = 0; i < n; i++) {
                vector<pair<unsigned int, unsigned int>> factor_status;
                for(auto [factor, count]: factors[i]) {
                    factor_status.emplace_back(factor, k - count);
                }
                ans += factor_status_count[factor_status];
                ++factor_status_count[factors[i]];
            }
            cout << ans;
            return 0;
        }

被这些函数引用 TEST().

TEST()

acwing::acwing4319::TEST	(	acwing4319	,
		case1
	)

在文件 acwing_test.cpp 第 2790 行定义.

                                {
            istringstream in("6 3\n1 3 9 8 24 1");
            auto out = ostringstream();
            main(in, out);
            const auto ans = out.str();
            ASSERT_EQ("5", ans);
        }

引用了 main().