Yandex.Algorithm 2014 Round 3 C問題 - Intervals

Source

Yandex.Algorithm 2014
Yandex.Algorithm 2014 Round 3
問題文

問題概要

要素数 $N$ の整数列 $\{A_k\}_{k=1}^N$ が与えられる.
その連続する部分列で,その部分列の中に含まれる $2$ つの要素の差が $D$ となるものがあるような,連続する部分列の数を求める問題.

解法

条件を満たさないもの,つまり,差が $D$ となる要素を含まないものを尺取法で数える.
$D=0$ のケースに注意.
時間計算量 $O(N)$.

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

template<class T>
struct ullHash{
  ull *key; T *val; unsigned memory, size, mask;

  void clear(){memset(val,0,size*sizeof(T));}
  void clear(int sz){size=1;while(size<sz)size*=2;mask=size-1;clear();}
  void init(int mem, int sz){memory=mem;size=1;while(size<sz)size*=2;mask=size-1;if(memory<size)memory=size;key=(ull*)malloc(memory*sizeof(ull));val=(T*)malloc(memory*sizeof(T));if(size)clear();}
  int function(ull a){return (a*97531)&mask;}
  T get(ull a){int i=function(a);for(;key[i]!=a&&val[i]!=0;)i=(i+1)&mask;if(key[i]!=a) return 0;return val[i];}
  void set(ull a, T v){int i=function(a);for(;key[i]!=a&&val[i]!=0;)i=(i+1)&mask;key[i]=a;val[i]=v;}
  T increase(ull a, T v = 1){int i=function(a);for(;key[i]!=a&&val[i]!=0;)i=(i+1)&mask;key[i]=a;return val[i]+=v;}
};

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(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);}
void writer(ll x, char c){int i,sz=0,m=0;char buf[20];if(x<0)m=1,x=-x;while(x)buf[sz++]=x%10,x/=10;if(!sz)buf[sz++]=0;if(m)mypc('-');while(sz--)mypc(buf[sz]+'0');mypc(c);}

int N, D;
ll A[1000000];

int main(){
  int i, j, k;
  int st;
  ll res;
  ullHash<int> hs;

  reader(&N,&D);
  rep(i,N) reader(A+i), A[i] += 1000000000000LL;
  
  hs.init(2*N, 2*N);
  res = (ll)N*(N+1) / 2;

  rep(i,N) hs.set(A[i], 1);

  st = 0;
  rep(i,N){
    while(hs.get(A[i]-D) > 1 || hs.get(A[i]+D) > 1){
      hs.increase(A[st], -1);
      st++;
    }
    res -= (i-st+1);

    hs.increase(A[i], 1);
  }

  writer(res, '\n');

  return 0;
}

Current time: 2017年11月21日04時14分33秒
Last modified: 2014年08月06日01時24分34秒 (by laycrs)
Tags: Competitive_Programming Yandex_Algorithm Yandex_Algorithm_2014 Yandex_Algorithm_2014_Round3
トップページに戻る

Logged in as: unknown user (not login)

ログイン: