結果

問題 No.12 限定された素数
コンテスト
ユーザー monnu
提出日時 2021-07-13 21:58:15
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 148 ms / 5,000 ms
コード長 1,701 bytes
コンパイル時間 6,207 ms
コンパイル使用メモリ 230,460 KB
実行使用メモリ 29,784 KB
最終ジャッジ日時 2024-07-02 17:16:10
合計ジャッジ時間 8,688 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll=long long;
#define MOD 998244353
using Graph=vector<vector<int>>;
#define INF 1000000000
#define MAX 5000000

vector<int> p;
vector<bool> p_table(MAX+1,true);
void prime(){
  p_table.at(0)=false,p_table.at(1)=false;
  for(int i=2;i<=MAX;i++){
    if(p_table.at(i)){
      p.push_back(i);
      for(int j=2;i*j<=MAX;j++){
        p_table.at(i*j)=false;
      }
    }
  }
}

int main(){
  int N;
  cin>>N;
  vector<bool> A(10,false);
  for(int i=0;i<N;i++){
    int a;
    cin>>a;
    A[a]=true;
  }

  prime();
  int n=p.size();
  vector<vector<int>> cnt(n+1,vector<int>(10,0));
  for(int i=0;i<n;i++){
    int x=p[i];
    for(int j=0;j<10;j++){
      cnt[i+1][j]+=cnt[i][j];
    }
    while(x>0){
      cnt[i+1][x%10]++;
      x/=10;
    }
  }

  int ans=-1;
  for(int i=0;i<n;i++){
    int left=i+1;
    int right=n+1;
    while(left+1<right){
      int x=(left+right)/2;
      int count=0;
      for(int j=0;j<10;j++){
        if(cnt[x][j]-cnt[i][j]>0){
          count++;
        }
      }
      if(count<=N){
        left=x;
      }else{
        right=x;
      }
    }
    bool flag=true;
    int x=left;
    for(int j=0;j<10;j++){
      if(A[j]==true){
        if(cnt[x][j]-cnt[i][j]==0){
          flag=false;
        }
      }else{
        if(cnt[x][j]-cnt[i][j]>0){
          flag=false;
        }
      }
    }
    if(flag==true){
      if(i==0&&x==n){
        ans=max(ans,5000000-1);
      }else if(i==0){
        ans=max(ans,p[x]-1-1);
      }else if(x==n){
        ans=max(ans,5000000-p[i-1]-1);
      }else{
        ans=max(ans,p[x]-1-p[i-1]-1);
      }
    }
  }
  cout<<ans<<endl;
}
0