結果

問題 No.1711 Divide LCM
ユーザー chineristACchineristAC
提出日時 2021-08-07 21:31:32
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 3,730 bytes
コンパイル時間 13,416 ms
コンパイル使用メモリ 296,644 KB
実行使用メモリ 4,348 KB
最終ジャッジ日時 2023-10-17 19:47:42
合計ジャッジ時間 20,491 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
testcase_38 RE -
testcase_39 RE -
testcase_40 RE -
testcase_41 RE -
testcase_42 RE -
testcase_43 RE -
testcase_44 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

#include <iostream>
#include <vector>
#include <string>
#include <map>
#include <set>
#include <queue>
#include <algorithm>
#include <cmath>
#include <iomanip>
#include <random>
#include <stdio.h>
#include <fstream>
#include <functional>
#include <cassert>
#include "testlib.h"
using namespace std;

#define rep(i,n,c) for (int i=0;i<n;i+=c)
#define append push_back
#define all(x) (x).begin(), (x).end()

template<class T>
using vec = vector<T>;
template<class T>
using vvec = vec<vec<T>>;
template<class T>
using vvvec = vec<vvec<T>>;
using ll = long long;
using pii = pair<int,int>;
using pll = pair<ll,ll>;


template<class T>
bool chmin(T &a, T b){
  if (a>b){
    a = b;
    return true;
  }
  return false;
}

template<class T>
bool chmax(T &a, T b){
  if (a<b){
    a = b;
    return true;
  }
  return false;
}

template<class T>
T sum(vec<T> x){
  T res=0;
  for (auto e:x){
    res += e;
  }
  return res;
}

template<class T>
void printv(vec<T> x){
  for (auto e:x){
    cout<<e<<" ";
  }
  cout<<"\n";
}

const int M = 1000000000;

int spf[32000];

void calc_spf(){
  spf[1] = 1;
  for (int i=2;i<32000;i++){
    if (spf[i]==0){
      spf[i] = i;
      for (int j=2*i;j<32000;j+=i){
        spf[j] = i;
      }
    }
  }
}

vec<int> prime = {};

void make_prime(){
  calc_spf();
  for (int i=2;i<32000;i++){
    if (spf[i]==i){
      prime.append(i);
    }
  }
}

vec<pll> factorization(ll n){
  assert (1 <= n);
  assert (n <= M);
  vec<pll> res;
  ll cnt = 0;
  for (auto p:prime){
    cnt = 0;
    while (n%p==0){
      cnt += 1;
      n /= p;
    }
    if (cnt!=0){
      res.append({p,cnt});
    }
  }
  if (n!=1){
    res.append({n,1});
  }
  return res;
}

vec<ll> divisors(vec<pll> n_prime){
  vec<ll> res = {1ll};
  ll tmp;
  for (auto [p,cnt]:n_prime){
    vec<ll> new_res;
    for (auto d:res){
      tmp = d;
      for (int i=0;i<cnt+1;i++){
        new_res.append(tmp);
        tmp *= p;
      }
    }
    res = new_res;
  }
  return res;
}

ll pow(ll n,ll k){
  ll res = 1;
  while (k){
    if (k&1){
      res *= n;
    }
    n *= n;
    k >>= 1;
  }
  return res;
}

int main(int argc, char* argv[]){
  registerValidation(argc,argv);
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);

  make_prime();

  int N;
  N = inf.readInt(1,2000);
  inf.readEoln();
  vec<ll> A(N);
  map<ll,ll> lcm_A;
  set<ll> all_divisors;
  set<ll> check;
  for (int i=0;i<N;i++){
    A[i] = inf.readLong(1ll,1000000000ll);
    check.insert(A[i]);
    auto a_prime = factorization(A[i]);
    for (auto [p,cnt]:a_prime){
      lcm_A[p];
      chmax(lcm_A[p],cnt);
    }
    auto D = divisors(a_prime);
    for (auto d:D){
      all_divisors.insert(d);
    }

    if (i==N-1){
      inf.readEoln();
    }
    else{
      inf.readSpace();
    }
  }

  inf.readEof();

  assert (check.size()==N);

  vec<ll> P(0);
  for (auto [p,cnt]:lcm_A){
    P.append(pow(p,cnt));
  }
  int n = P.size();

  deque<pll> deq = {{1,-1}};
  ll prod = -1;
  while (!deq.empty()){
    auto [v,k] = deq.front();
    deq.pop_front();
    for (int i=k+1;i<n;i++){
      if (!all_divisors.count(v*P[i])){
        prod = v*P[i];
        deq = {};
        break;
      }
      deq.append({v*P[i],i});
    }
  }

  if (prod==-1){
    cout<<-1<<endl;
    return 0;
  }

  vec<ll> selected_P(0);
  for (auto p:P){
    if (prod%p==0){
      selected_P.append(p);
    }
  }

  int k = selected_P.size();
  vec<vec<ll>> S(k,vec<ll>(0));
  for (auto a:A){
    for (int i=0;i<k;i++){
      if (a%selected_P[i]){
        S[i].append(a);
        break;
      }
    }
  }

  cout<<k<<endl;
  for (int i=0;i<k;i++){
    cout<<S[i].size()<<" ";
    printv(S[i]);
  }
}
0