結果

問題 No.368 LCM of K-products
ユーザー siguma323
提出日時 2016-05-03 22:28:44
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,776 bytes
コンパイル時間 936 ms
コンパイル使用メモリ 98,168 KB
実行使用メモリ 19,712 KB
最終ジャッジ日時 2024-10-05 06:01:58
合計ジャッジ時間 4,651 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <numeric>
#include <string>
#include <vector>
#include <iostream>
#include <queue>
#include <utility>
#include <cmath>
#include <map>
#include <unordered_map>
#include <functional>
using namespace std;

using ull = unsigned long long;

ull fast_pow(ull num,ull n)
{
    ull x = num;
    ull result = 1;
    while(n)
    {
        if(n & 1) result *= x;
        x *= x;
        n = n >> 1;
    }

    return result;
}

int main()
{
    ull n,k;
    cin >> n >> k;

    vector<ull> a(n);
    map<ull,vector<ull>> dic;

    for(int i =0; i < n; ++i)
    {
        cin >> a.at(i);
        ull tmp = a.at(i);

        int j = 2;
        while(j <= sqrt(a.at(i)))
        {
            if(tmp % j ==0)
            {
                vector<ull> t(n,0);
                tmp /= j;
                if(dic.find(j) != dic.end())
                {
                    dic.at(j).at(i) +=1;
                }
                else
                {
                    dic.emplace(j,t);
                    dic.at(j).at(i) = 1;
                }
            }
            else
            {
                ++j;
            }
        }

        vector<ull> t(n,0);
        if(dic.find(tmp) != dic.end())
        {
            dic.at(tmp).at(i) +=1;
        }
        else
        {
            dic.emplace(tmp,t);
            dic.at(tmp).at(i) = 1;
        }
    }

    ull sum = 1;
    for(auto&& x : dic)
    {
        sort(x.second.begin(),x.second.end(),greater<ull>());
        ull max_index = k > x.second.size() ? x.second.size() : k;
        ull count = 0;
        for(int i =0; i < k ; ++i)
        {
            count += x.second.at(i);
        }
        sum = (sum*fast_pow(x.first,count)) % static_cast<int>(1e9+7);
    }

    cout << sum << endl;
}
0