본문 바로가기

Java/Java 알고리즘 SWEA

[SW 아카데미] 2001. 파리 퇴치 #

반응형

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=2&contestProbId=AV5PzOCKAigDFAUq&categoryId=AV5PzOCKAigDFAUq&categoryType=CODE&problemTitle=&orderBy=RECOMMEND_COUNT&selectCodeLang=JAVA&select-1=2&pageSize=30&pageIndex=1 

 

SW Expert Academy

SW 프로그래밍 역량 강화에 도움이 되는 다양한 학습 컨텐츠를 확인하세요!

swexpertacademy.com

 

For문 만들기

-> 범위를 어떻게 지정할 것인가

 

-> 총 범위에서 원하는 범위를 지정하기 위해, 주어진 N과 M을 조합하는데

 

두 가지 방법으로 구현할 수 있다.

  1. M과 함께 움직이는 경우
  2. M이 따로 움직이는 경우

 

1. 함께 움직이는 경우

-> 총 범위와 지정하고 싶은 범위는 0부터 N-M까지로, i가 총범위인 N 끝까지 가지 않는다.

-> 움직이는 범위는 i부터 i+M까지 -> 옆으로 이동하고, 아래로 한칸 이동한다.

/////////////////////////////////////////////////////////////////////////////////////////////
// 기본 제공코드는 임의 수정해도 관계 없습니다. 단, 입출력 포맷 주의
// 아래 표준 입출력 예제 필요시 참고하세요.
// 표준 입력 예제
// int a;
// double b;
// char g;
// String var;
// long AB;
// a = sc.nextInt();                           // int 변수 1개 입력받는 예제
// b = sc.nextDouble();                        // double 변수 1개 입력받는 예제
// g = sc.nextByte();                          // char 변수 1개 입력받는 예제
// var = sc.next();                            // 문자열 1개 입력받는 예제
// AB = sc.nextLong();                         // long 변수 1개 입력받는 예제
/////////////////////////////////////////////////////////////////////////////////////////////
// 표준 출력 예제
// int a = 0;
// double b = 1.0;
// char g = 'b';
// String var = "ABCDEFG";
// long AB = 12345678901234567L;
//System.out.println(a);                       // int 변수 1개 출력하는 예제
//System.out.println(b); 		       						 // double 변수 1개 출력하는 예제
//System.out.println(g);		       						 // char 변수 1개 출력하는 예제
//System.out.println(var);		       				   // 문자열 1개 출력하는 예제
//System.out.println(AB);		       				     // long 변수 1개 출력하는 예제
/////////////////////////////////////////////////////////////////////////////////////////////
import java.util.Scanner;
import java.io.FileInputStream;

/*
   사용하는 클래스명이 Solution 이어야 하므로, 가급적 Solution.java 를 사용할 것을 권장합니다.
   이러한 상황에서도 동일하게 java Solution 명령으로 프로그램을 수행해볼 수 있습니다.
 */
class Solution
{
    public static void main(String args[]) throws Exception
    {
		/*
		   아래의 메소드 호출은 앞으로 표준 입력(키보드) 대신 input.txt 파일로부터 읽어오겠다는 의미의 코드입니다.
		   여러분이 작성한 코드를 테스트 할 때, 편의를 위해서 input.txt에 입력을 저장한 후,
		   이 코드를 프로그램의 처음 부분에 추가하면 이후 입력을 수행할 때 표준 입력 대신 파일로부터 입력을 받아올 수 있습니다.
		   따라서 테스트를 수행할 때에는 아래 주석을 지우고 이 메소드를 사용하셔도 좋습니다.
		   단, 채점을 위해 코드를 제출하실 때에는 반드시 이 메소드를 지우거나 주석 처리 하셔야 합니다.
		 */
       // System.setIn(new FileInputStream("src/sw2001/input.txt"));

		/*
		   표준입력 System.in 으로부터 스캐너를 만들어 데이터를 읽어옵니다.
		 */
        Scanner sc = new Scanner(System.in);
        int T;
        T=sc.nextInt();
		/*
		   여러 개의 테스트 케이스가 주어지므로, 각각을 처리합니다.
		*/

        for(int test_case = 1; test_case <= T; test_case++) {

            //초기화
            int n = sc.nextInt();
            int m=sc.nextInt();
            int[][] arr = new int[n][n];


            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    arr[i][j] = sc.nextInt();
                }
            }
            int sum=0;
            int answer=Integer.MIN_VALUE;


                //i: 0 0 1 1 0 0 1 1 0 0 1 1 0 0 1 1
                //j: 0 1 0 1 1 2 1 2 2 3 2 3 3 4 3 4

                //i: 1 1 2 2 1 1 2 2 1 1 2 2 1 1 2 2
                //j: 0 1 0 1 1 2 1 2 2 3 2 3 3 4 3 4
            for(int i=0;i<=n-m;i++){
                for(int j=0;j<=n-m;j++){
                    sum=0;
                    for(int k=i;k<i+m;k++){
                        for(int f=j;f<j+m;f++){
                          sum+=arr[k][f];
                        }
                    }
                    answer=Math.max(sum, answer);
                }
            }

            System.out.println("#"+test_case+" "+answer);

            /////////////////////////////////////////////////////////////////////////////////////////////
			/*
				 이 부분에 여러분의 알고리즘 구현이 들어갑니다.
			 */
            /////////////////////////////////////////////////////////////////////////////////////////////

        }
    }
}

2. 따로 움직이는 경우

: i와 j의 반복에 따라 ->M이 따로 움직이면서 배열의 값을 가져오기위해서 arr[i+a][j+b]로 값을 가져온다.

-> i와 j는 0부터 n-m까지 반복한다.

 

n:5 m:2일 경우

i와 j의 범위 : for(int i=0; i<=n-m; i++), for(int j=0; j<=n-m; j++) : 0, 1, 2,3로 총 범위에서 M의 범위를 뺀 범위만 돌아야한다.

-> M의 범위만큼 각각 돌아야 하기때문에, 빼줘야한다.

 

이후, M의 범위를 설정한다.

-> a와 b의 범위 : for(int a=0; a<M; a++), for(int b=0; b<M; b++) : 0, 1로 M이 따로 움직이면서, 배열의 값을 가져온다.

-> arr[i+a][j+b]


/////////////////////////////////////////////////////////////////////////////////////////////
// 기본 제공코드는 임의 수정해도 관계 없습니다. 단, 입출력 포맷 주의
// 아래 표준 입출력 예제 필요시 참고하세요.
// 표준 입력 예제
// int a;
// double b;
// char g;
// String var;
// long AB;
// a = sc.nextInt();                           // int 변수 1개 입력받는 예제
// b = sc.nextDouble();                        // double 변수 1개 입력받는 예제
// g = sc.nextByte();                          // char 변수 1개 입력받는 예제
// var = sc.next();                            // 문자열 1개 입력받는 예제
// AB = sc.nextLong();                         // long 변수 1개 입력받는 예제
/////////////////////////////////////////////////////////////////////////////////////////////
// 표준 출력 예제
// int a = 0;
// double b = 1.0;
// char g = 'b';
// String var = "ABCDEFG";
// long AB = 12345678901234567L;
//System.out.println(a);                       // int 변수 1개 출력하는 예제
//System.out.println(b); 		       						 // double 변수 1개 출력하는 예제
//System.out.println(g);		       						 // char 변수 1개 출력하는 예제
//System.out.println(var);		       				   // 문자열 1개 출력하는 예제
//System.out.println(AB);		       				     // long 변수 1개 출력하는 예제
/////////////////////////////////////////////////////////////////////////////////////////////
import java.util.Scanner;
import java.io.FileInputStream;

/*
   사용하는 클래스명이 Solution 이어야 하므로, 가급적 Solution.java 를 사용할 것을 권장합니다.
   이러한 상황에서도 동일하게 java Solution 명령으로 프로그램을 수행해볼 수 있습니다.
 */
class Solution
{
    public static void main(String args[]) throws Exception
    {
		/*
		   아래의 메소드 호출은 앞으로 표준 입력(키보드) 대신 input.txt 파일로부터 읽어오겠다는 의미의 코드입니다.
		   여러분이 작성한 코드를 테스트 할 때, 편의를 위해서 input.txt에 입력을 저장한 후,
		   이 코드를 프로그램의 처음 부분에 추가하면 이후 입력을 수행할 때 표준 입력 대신 파일로부터 입력을 받아올 수 있습니다.
		   따라서 테스트를 수행할 때에는 아래 주석을 지우고 이 메소드를 사용하셔도 좋습니다.
		   단, 채점을 위해 코드를 제출하실 때에는 반드시 이 메소드를 지우거나 주석 처리 하셔야 합니다.
		 */
       // System.setIn(new FileInputStream("src/sw2001/input.txt"));

		/*
		   표준입력 System.in 으로부터 스캐너를 만들어 데이터를 읽어옵니다.
		 */
        Scanner sc = new Scanner(System.in);
        int T;
        T=sc.nextInt();
		/*
		   여러 개의 테스트 케이스가 주어지므로, 각각을 처리합니다.
		*/

        for(int test_case = 1; test_case <= T; test_case++) {

            //초기화
            int n = sc.nextInt();
            int m=sc.nextInt();
            int[][] arr = new int[n][n];


            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    arr[i][j] = sc.nextInt();
                }
            }
            int sum=0;
            int answer=Integer.MIN_VALUE;


                //i: 0 0 1 1 0 0 1 1 0 0 1 1 0 0 1 1
                //j: 0 1 0 1 1 2 1 2 2 3 2 3 3 4 3 4

                //i: 1 1 2 2 1 1 2 2 1 1 2 2 1 1 2 2
                //j: 0 1 0 1 1 2 1 2 2 3 2 3 3 4 3 4
            //n=5; m=3일 경우
            //i와 j는 0~2를 돈다.
            //i=0 j=0,1,2 ->  0~2, 1~3, 2~4
            //i=1
            //i=2
            
            //m : 0,1,2 -> 1, 2, 3 -> 2, 3, 4
            for(int i=0;i<=n-m;i++){
                for(int j=0;j<=n-m;j++){
                    sum=0;
                    //a: 0,1,2
                    for(int a=0; a<m; a++){
                        for(int b=0; b<m; b++){
                            sum+=arr[i+a][j+b];
                        }
                    }
                    answer=Math.max(answer,sum);
                }
            }
            System.out.println("#"+test_case+" "+answer);

            /////////////////////////////////////////////////////////////////////////////////////////////
			/*
				 이 부분에 여러분의 알고리즘 구현이 들어갑니다.
			 */
            /////////////////////////////////////////////////////////////////////////////////////////////

        }
    }
}
반응형