fork download
  1. -- CodeIQ Q3434 ストレート・ラインズ
  2. -- author: Leonardone @ NEETSDKASU
  3.  
  4. import Data.List (foldl')
  5. g n k = takeWhile ((<n).(*k)) $ dropWhile ((<=(-n)).(*k)) [(-n)..n]
  6. 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]
  7. f n = (`div`2) $ (h n 3)-2*(h n 2)+(h n 1)
  8. main = print $ f (10^7)
Time limit exceeded #stdin #stdout 5s 8388607KB
stdin
Standard input is empty
stdout
Standard output is empty