[SQL] [Codewars] 6Kyu - SELECT prime numbers

Codewars: Achieve mastery through coding challenge

문제

Write a SELECT query which will return all prime numbers smaller than 100 in ascending order.

Your query should return one column named prime.

 

코드

CREATE OR REPLACE FUNCTION pri(n INTEGER)
RETURNS INTEGER AS $$
DECLARE
    i INTEGER := 1;
    cnt INTEGER := 0;
BEGIN
    FOR i IN 1..100
    LOOP
        i := i+1;
          if n % i = 0 and n != i THEN
            BEGIN
              cnt := cnt + 1;
              EXIT;
            END;
          END IF;
    END LOOP;
    RETURN cnt;
END;
$$ LANGUAGE plpgsql;

CREATE TABLE my_table(
  list INTEGER PRIMARY KEY
);

INSERT INTO my_table SELECT j FROM generate_series(2,100) j;

SELECT list as prime FROM my_table WHERE pri(list) = 0;

함수를 만들어서 풀었는데…

함수로 풀지 말라며 문구가 떴다.

SELECT i as prime
FROM
  generate_series(2, 100) as i
WHERE
  i IN (2,3,5,7) OR
  i % 2 != 0 and
  i % 3 != 0 and
  i % 5 != 0 and
  i % 7 != 0

 

설명

좀 원시적인 풀이지만, 100까지의 자연수에서 소수를 찾기만 하면 되는 것이므로, 2, 3, 5, 7로 나누어 떨어지지 않는 수를 골라냈다.