#362426 (C++17) No.854 公平なりんご分配

提出ソース

結果

問題	No.854 公平なりんご分配
ユーザー	beet
提出日時	2019-07-26 21:43:10
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	2,236 ms / 3,153 ms
コード長	4,479 bytes
コンパイル時間	2,718 ms
コンパイル使用メモリ	230,660 KB
実行使用メモリ	47,608 KB
最終ジャッジ日時	2024-07-02 06:54:17
合計ジャッジ時間	14,156 ms
ジャッジサーバーID （参考情報）	judge5 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	3 ms 7,120 KB
testcase_01	AC	3 ms 7,092 KB
testcase_02	AC	3 ms 7,132 KB
testcase_03	AC	3 ms 7,248 KB
testcase_04	AC	3 ms 7,208 KB
testcase_05	AC	3 ms 7,076 KB
testcase_06	AC	3 ms 7,068 KB
testcase_07	AC	3 ms 7,276 KB
testcase_08	AC	4 ms 7,248 KB
testcase_09	AC	3 ms 7,108 KB
testcase_10	AC	4 ms 7,232 KB
testcase_11	AC	4 ms 7,176 KB
testcase_12	AC	4 ms 7,140 KB
testcase_13	AC	3 ms 7,148 KB
testcase_14	AC	4 ms 7,168 KB
testcase_15	AC	3 ms 7,108 KB
testcase_16	AC	4 ms 6,964 KB
testcase_17	AC	4 ms 7,132 KB
testcase_18	AC	4 ms 7,224 KB
testcase_19	AC	3 ms 7,168 KB
testcase_20	AC	4 ms 7,188 KB
testcase_21	AC	3 ms 7,088 KB
testcase_22	AC	6 ms 7,392 KB
testcase_23	AC	5 ms 7,296 KB
testcase_24	AC	6 ms 7,468 KB
testcase_25	AC	4 ms 7,296 KB
testcase_26	AC	7 ms 7,480 KB
testcase_27	AC	5 ms 7,600 KB
testcase_28	AC	5 ms 7,320 KB
testcase_29	AC	4 ms 7,384 KB
testcase_30	AC	5 ms 7,260 KB
testcase_31	AC	6 ms 7,496 KB
testcase_32	AC	36 ms 15,232 KB
testcase_33	AC	43 ms 11,776 KB
testcase_34	AC	81 ms 18,672 KB
testcase_35	AC	56 ms 15,872 KB
testcase_36	AC	45 ms 8,320 KB
testcase_37	AC	36 ms 14,592 KB
testcase_38	AC	26 ms 13,072 KB
testcase_39	AC	126 ms 19,784 KB
testcase_40	AC	54 ms 11,264 KB
testcase_41	AC	56 ms 14,464 KB
testcase_42	AC	74 ms 18,256 KB
testcase_43	AC	91 ms 15,232 KB
testcase_44	AC	91 ms 17,792 KB
testcase_45	AC	97 ms 10,288 KB
testcase_46	AC	123 ms 18,176 KB
testcase_47	AC	45 ms 12,800 KB
testcase_48	AC	69 ms 18,080 KB
testcase_49	AC	63 ms 18,056 KB
testcase_50	AC	30 ms 14,464 KB
testcase_51	AC	118 ms 18,404 KB
testcase_52	AC	96 ms 12,788 KB
testcase_53	AC	48 ms 11,904 KB
testcase_54	AC	91 ms 13,564 KB
testcase_55	AC	30 ms 11,884 KB
testcase_56	AC	26 ms 11,776 KB
testcase_57	AC	45 ms 11,336 KB
testcase_58	AC	95 ms 9,856 KB
testcase_59	AC	23 ms 10,612 KB
testcase_60	AC	63 ms 11,520 KB
testcase_61	AC	35 ms 8,320 KB
testcase_62	AC	72 ms 11,264 KB
testcase_63	AC	60 ms 11,228 KB
testcase_64	AC	20 ms 8,576 KB
testcase_65	AC	50 ms 16,636 KB
testcase_66	AC	42 ms 9,980 KB
testcase_67	AC	71 ms 13,824 KB
testcase_68	AC	76 ms 10,720 KB
testcase_69	AC	46 ms 18,524 KB
testcase_70	AC	26 ms 11,040 KB
testcase_71	AC	30 ms 11,324 KB
testcase_72	AC	62 ms 8,704 KB
testcase_73	AC	40 ms 14,828 KB
testcase_74	AC	88 ms 17,280 KB
testcase_75	AC	44 ms 11,648 KB
testcase_76	AC	75 ms 15,204 KB
testcase_77	AC	77 ms 18,304 KB
testcase_78	AC	92 ms 11,252 KB
testcase_79	AC	86 ms 14,384 KB
testcase_80	AC	81 ms 14,576 KB
testcase_81	AC	62 ms 14,700 KB
testcase_82	AC	207 ms 47,456 KB
testcase_83	AC	286 ms 47,580 KB
testcase_84	AC	2,236 ms 47,608 KB
testcase_85	AC	259 ms 45,524 KB
testcase_86	AC	142 ms 45,660 KB
testcase_87	AC	353 ms 47,580 KB
testcase_88	AC	335 ms 47,580 KB
testcase_89	AC	330 ms 47,572 KB
testcase_90	AC	338 ms 47,536 KB
testcase_91	AC	341 ms 47,580 KB
testcase_92	AC	413 ms 47,540 KB
testcase_93	AC	427 ms 47,476 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include<bits/stdc++.h>
using namespace std;
using Int = long long;
//BEGIN CUT HERE
struct Mo{
  using F = function<void(Int)>;
  vector<Int> ls,rs,ord;
  Int n,width,nl,nr,ptr;
  vector<bool> flg;
  F expand,shrink;

  Mo(Int n,Int width,F expand,F shrink):
    n(n),width(width),nl(0),nr(0),ptr(0),flg(n,0),
    expand(expand),shrink(shrink){}

  void add(Int l,Int r){
    ls.emplace_back(l);
    rs.emplace_back(r);
  }

  void build(){
    ord.resize(ls.size());
    iota(ord.begin(),ord.end(),0);
    sort(ord.begin(),ord.end(),
         [&](Int a,Int b){
           if(ls[a]/width!=ls[b]/width) return ls[a]<ls[b];
           return bool((rs[a]<rs[b])^((ls[a]/width)&1));
         });
  }

  Int process(){
    if(ptr==(Int)ord.size()) return -1;
    const Int idx=ord[ptr++];
    while(nl>ls[idx]) calc(--nl);
    while(nr<rs[idx]) calc(nr++);
    while(nl<ls[idx]) calc(nl++);
    while(nr>rs[idx]) calc(--nr);
    return idx;
  }

  inline void calc(Int idx){
    flg[idx].flip();
    if(flg[idx]) expand(idx);
    else         shrink(idx);
  }

};
//END CUT HERE

template<typename T>
map<T, Int> factorize(T x){
  map<T, Int> res;
  for(Int i=2;i*i<=x;i++){
    while(x%i==0){
      x/=i;
      res[i]++;
    }
  }
  if(x!=1) res[x]++;
  return res;
}


template<typename T>
vector<T> make_v(size_t a){return vector<T>(a);}
template<typename T>
vector<vector<T> > make_v(size_t a,size_t b){
  return vector<vector<T> >(a,make_v<T>(b));
}
template<typename T>
vector<vector<vector<T> > > make_v(size_t a,size_t b,size_t c){
  return vector<vector<vector<T> > > (a,make_v<T>(b,c));
}

template<typename T,typename V>
typename enable_if<is_class<T>::value==0>::type
fill_v(T &t,const V &v){t=v;}

template<typename T,typename V>
typename enable_if<is_class<T>::value!=0>::type
fill_v(T &t,const V &v){
  for(auto &e:t) fill_v(e,v);
}


class EulerTourForEdge{
private:
  vector<Int> ds,us,dep,btm;

  void dfs(Int v,Int p,Int d){
    dep[v]=d;
    for(Int u:G[v]){
      if(u==p) continue;
      ds[u]=btm.size();
      btm.emplace_back(u);
      dfs(u,v,d+1);
      us[u]=btm.size();
      btm.emplace_back(u);
    }
  }
public:
  vector<vector<Int> > G;

  EulerTourForEdge(){}
  EulerTourForEdge(Int n):
    ds(n),us(n),dep(n),G(n){}

  void add_edge(Int u,Int v){
    G[u].emplace_back(v);
    G[v].emplace_back(u);
  }

  void build(Int r=0){
    btm.clear();
    ds[r]=btm.size();
    btm.emplace_back(r);
    dfs(r,-1,0);
    us[r]=btm.size();
    btm.emplace_back(r);
  }

  Int child(Int u,Int v){
    return dep[u]<dep[v]?v:u;
  }

  Int bottom(Int e){
    return btm[e];
  }

  // lca(u, v) must be u or v
  template<typename F>
  void query(Int u,Int v,F f){
    if(dep[u]>dep[v]) swap(u,v);
    f(ds[u]+1,ds[v]+1);
  }

  template<typename T,typename G>
  void update(Int v,T x,G g){
    g(ds[v], x);
    g(us[v],-x);
  }
};

//INSERT ABOVE HERE
signed DWANGO2017FINAL_B(){
  using ll = long long;
  Int n;
  cin>>n;
  vector<Int> x(n);
  for(Int i=0;i<n;i++) cin>>x[i];

  const Int RT = 40;
  auto acc=make_v<Int>(RT,n+1);
  fill_v(acc,0);

  using P = pair<Int, Int>;
  vector<vector<P> > v(n);
  for(Int i=0;i<n;i++){
    for(auto p:factorize(x[i])){
      if(p.first<RT) acc[p.first][i+1]+=p.second;
      else v[i].emplace_back(p);
    }
  }

  for(Int j=0;j<RT;j++)
    for(Int i=0;i<n;i++)
      acc[j][i+1]+=acc[j][i];

  const Int MAX = 5e5+100;
  vector<Int> cnt(MAX,0);

  auto expand=[&](Int idx){
                for(auto p:v[idx]){
                  cnt[p.first]+=p.second;
                }
              };
  auto shrink=[&](Int idx){
                for(auto p:v[idx]){
                  cnt[p.first]-=p.second;
                }
              };
  Mo mo(n,400,expand,shrink);

  Int q;
  cin>>q;
  vector<Int> ps(q);
  for(Int i=0;i<q;i++){
    Int l,r;
    cin>>ps[i]>>l>>r;
    l--;
    mo.add(l,r);
  }
  mo.build();

  vector<ll> ans(q,1);
  for(Int i=0;i<q;i++){
    Int k=mo.process();
    Int l=mo.ls[k],r=mo.rs[k];
    if((acc[0][r]-acc[0][l])>=1){
      ans[k]=1;
      continue;
    }
    auto mp=factorize(ps[k]);
    for(auto p:mp){
      if(p.first<RT) ans[k]&=(acc[p.first][r]-acc[p.first][l])>=p.second;
      else ans[k]&=cnt[p.first]>=p.second;
    }
  }

  for(Int i=0;i<q;i++) cout<<(ans[i]?"Yes":"NO")<<"\n";
  cout<<flush;
  return 0;
}
/*
  verified on 2018/02/02
  https://beta.atcoder.jp/contests/dwacon2017-honsen/tasks/dwango2017final_b
*/



signed main(){
  DWANGO2017FINAL_B();
  //ABC133_F();
  return 0;
}

yukicoder

結果

テストケース

ソースコード