#687321 (C++17) No.1711 Divide LCM

• 問題: No.1711 Divide LCM
• ユーザー: chineristAC
• 提出日時: 2021-08-07 21:10:01
• 言語: C++17 (gcc 15.2.0 + boost 1.90.0)
• 結果: RE
• 実行時間: -
• コード長: 3,750 bytes
• コンパイル時間: 9,003 ms
• コンパイル使用メモリ: 355,524 KB
• 実行使用メモリ: 7,972 KB
• 最終ジャッジ日時: 2026-06-22 12:32:18
• 合計ジャッジ時間: 17,928 ms
• ジャッジサーバーID （参考情報）: judge1_0 / judge2_1

ファイルパターン	結果
sample	RE * 3
other	RE * 42

ソースコード

#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,cnt});
  }
  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.find(v*P[i])==all_divisors.end()){
        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]);
  }
}