백준 15829번

50점이 나와서 헤맸던 문제이다.

 

기존에 제출했던 코드이다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class M15829 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int r = 31;
        int M = 1234567891;

        int N = Integer.parseInt(st.nextToken());

        String[] strings = br.readLine().split("");

        int result = 0;
        for (int i = 0; i< strings.length; i++){
            int code = (int)strings[i].charAt(0)-96;
            result += (int) (code * Math.pow(r, i));
        }

        System.out.println(result % M);
    }
}

변숫값의 오버플로우,  나머지 값을 구하는 과정에서의 오버플로우를 고려하지 못하여 한번에 풀지 못했다!!

 

 

먼저 결과값의 변수형을 long으로 고쳐주었다. 

그리고 Math.pow를 쓸 때 long 범위를 초과하는 경우가 있기 때문에 모듈러 연산을 이용했다.

모듈러 연산 중 분배법칙을 사용하면 overflow를 해결할 수 있다. -> 큰 수로 나머지를 구하라는 문제에서, 연산 과정 중에서도 나머지 연산을 취하는 경우가 많음! 

모듈러 연산 => (A B) mod C = (A mod C B mod C) mod C

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class M15829 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());

        String[] strings = br.readLine().split("");

        int r = 31;
        int M = 1234567891;
        long pow = 1;
        long result = 0;
        
        for (String string : strings) {
            int code = (int) ((string.charAt(0) - 96) * pow % M);
            pow = pow * r % M;
            result += code;
        }

        System.out.println(result % M);
    }
}

 

 

 

쉬워보이는 문제였는데, 대충 풀다가 틀렸다! 조금 더 꼼꼼하게 생각하는 습관을 가지도록 하자~