fork download
  1. -- CodeIQ Q3434 ストレート・ラインズ
  2. -- author: Leonardone @ NEETSDKASU
  3.  
  4. g n k = takeWhile ((<n).(*k)) $ dropWhile ((<=(-n)).(*k)) [(-n)..n]
  5. h n k = foldl (+) 0 [(n-k*abs x)*(n-k*abs y)|x<-g n k,y<-g n k,gcd x y==1]
  6. f n = (`div`2) $ (h n 3)-2*(h n 2)+(h n 1)
  7. main = print $ f (10^3)
Success #stdin #stdout 0.37s 8388607KB
stdin
Standard input is empty
stdout
185755065672