yukicoder No.12 - 限定された素数

Source

ニコニコミュニティ
問題文

問題概要

$0$ から $9$ までのそれぞれの数字に対して,使ってはいけない,か,使わなければいけない,のどちらかが与えられる.
$1 \leq K \leq L \leq 5000000\ (5 \times 10^6)$ なる範囲 $[K,L]$ で,$K$ 以上 $L$ 以下の素数を列挙した時,使ってはいけない数字は含まれておらず,使わなければいけない数字はどこかに含まれているような $K,L$ のうち,$L-K$ の最大値を求める問題.
そのような区間 $[K,L]$ が存在しないならそれを指摘する.

解法

素数をエラトステネスの篩などで求めておく.
どの数字が何回使われているかを数えながら尺取法すれば良い.

C++によるスパゲッティなソースコード

#include<bits/stdc++.h>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
void writer(int x, char c){int s=0,m=0;char f[10];if(x<0)m=1,x=-x;while(x)f[s++]=x%10,x/=10;if(!s)f[s++]=0;if(m)mypc('-');while(s--)mypc(f[s]+'0');mypc(c);}

int getPrime(int N, int res[]){int i,a,b,s=1,f[4780],S=1;const int r=23000;bool B[r],*p=B;N/=2;res[0]=2;b=min(r,N);REP(i,1,b)p[i]=1;REP(i,1,b)if(p[i]){res[s++]=2*i+1;f[S]=2*i*(i+1);if(f[S]<N){for(;f[S]<r;f[S]+=res[S])p[f[S]]=0;S++;}}for(a=r;a<N;a+=r){b=min(a+r,N);p-=r;REP(i,a,b)p[i]=1;REP(i,1,S)for(;f[i]<b;f[i]+=res[i])p[f[i]]=0;REP(i,a,b)if(p[i])res[s++]=2*i+1;}return s;}

#define LIM 5000001

int p[LIM], ps;
char chk[LIM];
int mask;
int nw, cnt[10];

int main(){
  int i, j, k;
  int st, ed;
  int res = -1;

  reader(&k);
  while(k--) reader(&i), mask |= 1<<i;

  ps = getPrime(LIM, p);
  rep(i,ps) chk[p[i]] = 1;

  st = ed = 1;
  while(ed < LIM){
    if(nw == mask) res = max(res, ed-st-1);
    if(nw <= mask){
      k = ed++; if(chk[k]==0) k = 0;
      while(k){
        cnt[k%10]++;
        if(cnt[k%10]==1) nw ^= (1<<(k%10));
        k /= 10;
      }
    }

    if(nw == mask) res = max(res, ed-st-1);
    if(nw > mask){
      k = st++; if(chk[k]==0) k = 0;
      while(k){
        cnt[k%10]--;
        if(!cnt[k%10]) nw ^= (1<<(k%10));
        k /= 10;
      }
    }
  }

  writer(res,'\n');

  return 0;
}

Current time: 2017年09月25日11時34分24秒
Last modified: 2014年11月11日23時59分44秒 (by laycrs)
Tags: Competitive_Programming yukicoder
トップページに戻る

Logged in as: unknown user (not login)

ログイン: