提出 #9587479

---

ソースコード 拡げる

#pragma GCC optimize ("Ofast")
#include<bits/stdc++.h>
using namespace std;
void *wmem;
char memarr[96000000];
template<class S, class T> inline S min_L(S a,T b){
  return a<=b?a:b;
}
template<class T> inline void walloc1d(T **arr, int x, void **mem = &wmem){
  static int skip[16] = {0, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1};
  (*mem) = (void*)( ((char*)(*mem)) + skip[((unsigned long long)(*mem)) & 15] );
  (*arr)=(T*)(*mem);
  (*mem)=((*arr)+x);
}
template<class T1> void sortA_L(int N, T1 a[], void *mem = wmem){
  sort(a, a+N);
}
template<class T1, class T2> void sortA_L(int N, T1 a[], T2 b[], void *mem = wmem){
  int i;
  pair<T1, T2> *arr;
  walloc1d(&arr, N, &mem);
  for(i=(0);i<(N);i++){
    arr[i].first = a[i];
    arr[i].second = b[i];
  }
  sort(arr, arr+N);
  for(i=(0);i<(N);i++){
    a[i] = arr[i].first;
    b[i] = arr[i].second;
  }
}
inline void rd(long long &x){
  int k;
  int m=0;
  x=0;
  for(;;){
    k = getchar_unlocked();
    if(k=='-'){
      m=1;
      break;
    }
    if('0'<=k&&k<='9'){
      x=k-'0';
      break;
    }
  }
  for(;;){
    k = getchar_unlocked();
    if(k<'0'||k>'9'){
      break;
    }
    x=x*10+k-'0';
  }
  if(m){
    x=-x;
  }
}
inline void wt_L(char a){
  putchar_unlocked(a);
}
inline void wt_L(int x){
  int s=0;
  int 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){
    putchar_unlocked('-');
  }
  while(s--){
    putchar_unlocked(f[s]+'0');
  }
}
inline void wt_L(long long x){
  int s=0;
  int 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){
    putchar_unlocked('-');
  }
  while(s--){
    putchar_unlocked(f[s]+'0');
  }
}
template<class T, class U> inline T GCD_L(T a, U b){
  T r;
  while(b){
    r=a;
    a=b;
    b=r%a;
  }
  return a;
}
template<class S> inline void arrInsert(const int k, int &sz, S a[], const S aval){
  int i;
  sz++;
  for(i=sz-1;i>k;i--){
    a[i] = a[i-1];
  }
  a[k] = aval;
}
template<class S, class T> inline void arrInsert(const int k, int &sz, S a[], const S aval, T b[], const T bval){
  int i;
  sz++;
  for(i=sz-1;i>k;i--){
    a[i] = a[i-1];
  }
  for(i=sz-1;i>k;i--){
    b[i] = b[i-1];
  }
  a[k] = aval;
  b[k] = bval;
}
template<class S, class T, class U> inline void arrInsert(const int k, int &sz, S a[], const S aval, T b[], const T bval, U c[], const U cval){
  int i;
  sz++;
  for(i=sz-1;i>k;i--){
    a[i] = a[i-1];
  }
  for(i=sz-1;i>k;i--){
    b[i] = b[i-1];
  }
  for(i=sz-1;i>k;i--){
    c[i] = c[i-1];
  }
  a[k] = aval;
  b[k] = bval;
  c[k] = cval;
}
template<class S, class T, class U, class V> inline void arrInsert(const int k, int &sz, S a[], const S aval, T b[], const T bval, U c[], const U cval, V d[], const V dval){
  int i;
  sz++;
  for(i=sz-1;i>k;i--){
    a[i] = a[i-1];
  }
  for(i=sz-1;i>k;i--){
    b[i] = b[i-1];
  }
  for(i=sz-1;i>k;i--){
    c[i] = c[i-1];
  }
  for(i=sz-1;i>k;i--){
    d[i] = d[i-1];
  }
  a[k] = aval;
  b[k] = bval;
  c[k] = cval;
  d[k] = dval;
}
template<class S, class T> inline S chmax(S &a, T b){
  if(a<b){
    a=b;
  }
  return a;
}
long long N;
long long X;
long long D;
int sz;
long long mns[200000];
long long mxs[200000];
int main(){
  wmem = memarr;
  long long i;
  long long j;
  long long g;
  long long mn;
  long long mx;
  long long all;
  long long res = 0;
  rd(N);
  rd(X);
  rd(D);
  if(X==D && D==0){
    wt_L(1);
    wt_L('\n');
    return 0;
  }
  if(D==0){
    wt_L(N+1);
    wt_L('\n');
    return 0;
  }
  if(D < 0){
    D = -D;
    X = -X;
  }
  g =GCD_L(abs(X), D);
  X /= g;
  D /= g;
  all = N * (N-1) / 2 * D;
  int Q5VJL1cS = min_L(D, N+1);
  for(i=(0);i<(Q5VJL1cS);i++){
    sz = 0;
    for(j=(i);j<(N+1);j+=(D)){
      mx = X * (N-j) + (all - j * (j-1) / 2 * D);
      mn = X * (N-j) + ((N-j) * (N-j-1) / 2 * D);
      arrInsert(sz,sz,mns,mn,mxs,mx);
    }
    sortA_L(sz, mns, mxs);
    mx = -4611686016279904256LL;
    for(j=(0);j<(sz);j++){
      chmax(mns[j], mx + D);
      chmax(mx, mxs[j]);
      if(mxs[j] >= mns[j]){
        res += (mxs[j] - mns[j]) / D + 1;
      }
    }
  }
  wt_L(res);
  wt_L('\n');
  return 0;
}
// cLay varsion 20200119-1

// --- original code ---
// ll N, X, D;
// 
// int sz;
// ll mns[2d5], mxs[2d5];
// 
// {
//   ll i, j, g, mn, mx, all;
//   ll res = 0;
// 
//   rd(N,X,D);
// 
//   if(X==D==0) wt(1), return 0;
//   if(D==0) wt(N+1), return 0;
// 
//   if(D < 0) D = -D, X = -X;
//   g = gcd(abs(X),D);
//   X /= g;
//   D /= g;
// 
//   all = N * (N-1) / 2 * D;
//   REP(i,min(D,N+1)){
//     sz = 0;
//     rep(j,i,N+1,D){
//       mx = X * (N-j) + (all - j * (j-1) / 2 * D);
//       mn = X * (N-j) + ((N-j) * (N-j-1) / 2 * D);
//       arrInsert(sz,sz,mns,mn,mxs,mx);
//     }
// 
//     sortA(sz, mns, mxs);
//     mx = -ll_inf;
//     rep(j,sz){
//       mns[j] >?= mx + D;
//       mx >?= mxs[j];
//       if(mxs[j] >= mns[j]) res += (mxs[j] - mns[j]) / D + 1;
//     }
//   }
// 
//   wt(res);
// }

提出情報

ジャッジ結果