CodeChef September 2014 Cook-Off 3問目 - Subarray LCM

Source

CodeChef September 2014 Cook-Off
問題文

問題概要

要素数 $N$ の正整数の列 $\{A_k\}_{k=1}^N$ が与えられる.
長さ $2$ 以上の連続する部分列 $\{A_l, A_{l+1}, \ldots, A_r\}$ で,LCMが $A_l \times A_{l+1} \times \cdots A_r$ となるもののうち,最大の長さのものの長さを求める問題.
そのようなものが存在しないならそれを指摘する.

解法

要するに,互いに同じ素因数を持たないような区間の長さの最大値を求める問題.
各整数 $A_k$ を素因数分解して,$A_k$ がもつ(異なる)素因数を列挙しておく.
後は,尺取法で求める.

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)

#define ll long long
#define ull unsigned ll

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 reader(int *x, int *y){reader(x);reader(y);}
void reader(int *x, int *y, int *z){reader(x);reader(y);reader(z);}
void reader(ll *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);}
int reader(char c[]){int i,s=0;for(;;){mygc(i);if(i!=' '&&i!='\n'&&i!='\r'&&i!='\t'&&i!=EOF) break;}c[s++]=i;for(;;){mygc(i);if(i==' '||i=='\n'||i=='\r'||i=='\t'||i==EOF) break;c[s++]=i;}return s;}

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);}
void writer(ll x, char c){int s=0,m=0;char f[20];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);}
void writer(const char c[]){int i;for(i=0;c[i]!='\0';i++)mypc(c[i]);}

template<class T> T GCD(T a,T b){T r; while(a){r=b; b=a; a=r%a;} return b;}
template<class T> T LCM(T a,T b){return a/GCD(a,b)*b;}

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;}
int ps, p[111111];
int pind[1111111];

int T, N, A[111111];
int app[111111];
int dame;

int fact[111111][20], factn[111111];

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

  ps = getPrime(1000000, p);
  rep(i,ps) pind[p[i]] = i;

  reader(&T);
  while(T--){
    reader(&N);
    rep(i,N) reader(A+i);

    rep(i,N){
      factn[i] = 0;
      rep(k,ps){
        if(p[k]*p[k] > A[i]) break;
        if(A[i]%p[k]==0){
          while(A[i]%p[k]==0) A[i] /= p[k];
          fact[i][factn[i]++] = k;
        }
      }
      if(A[i] > 1){
        fact[i][factn[i]++] = pind[A[i]];
      }
    }

    rep(i,ps) app[i] = 0;
    st = ed = 0;
    dame = 0;
    res = 0;
    while(ed < N){
      while(dame==0 && ed < N){
        rep(i,factn[ed]){
          j = fact[ed][i];
          if(app[j]==1) dame++;
          app[j]++;
        }
        ed++;
        if(dame==0) res = max(res, ed-st);
      }
      while(dame){
        rep(i,factn[st]){
          j = fact[st][i];
          if(app[j]==2) dame--;
          app[j]--;
        }
        st++;
        if(dame==0) res = max(res, ed-st);
      }
    }
    if(res<=1) res = -1;
    writer(res,'\n');
  }

  return 0;
}

Current time: 2024年03月19日19時27分33秒
Last modified: 2014年09月22日21時44分58秒 (by laycrs)
Tags: Competitive_Programming CodeChef CodeChef_CookOff CookOff_201409
トップページに戻る

Logged in as: unknown user (not login)

ログイン: