글 읽기 - 예제는 다 맞는데 왜 틀렸습니다!가 나오는지 모르겠습니다!!

seulshine 3년 전 0

안녕하세요 해당 문제 해결에 필요한 로직을 다 작성했다고 생각했는데 계속 틀렸습니다가 나와서 결국 질문 글을 올리게 되었습니다 ㅠㅠ

Java 에서 충돌?하는 파이어볼에 대한 로직은 HashSet 과 HashMap 을 사용하였습니다!

함께 틀리는 이유를 찾아봐주시면 정말 감사드리겠습니다...!!!!!

+ 원래 dead 라는 플래그를 사용하여 전체를 다 봤는데 시간초과가 나서 급하게 fireballs = newFireballs; 로 덮어씌우는 로직으로 수정했었습니다!

dead 플래그는 사용되지 않고 있습니다!

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.StringTokenizer;

public class Main_마법사와파이어볼 {
   
   private static int N;
   private static int M;
   private static int K;
   private static int[][] map;
   
   static class Fireball {
      int r;
      int c;
      int m; // 질량
      int s; // 속도
      int d; // 방향 
      
      boolean dead;
      // 7 0 1
      // 6   2
      // 5 4 3

      public int getR() {
         return r;
      }
      public Fireball(int r, int c, int m, int s, int d, boolean dead) {
         super();
         this.r = r;
         this.c = c;
         this.m = m;
         this.s = s;
         this.d = d;
         this.dead = dead;
      }
      public void setR(int r) {
         this.r = r;
      }
      public int getC() {
         return c;
      }
      public void setC(int c) {
         this.c = c;
      }
      public boolean isDead() {
         return dead;
      }
      public void setDead(boolean dead) {
         this.dead = dead;
      }
      @Override
      public String toString() {
         return "Fireball [r=" + r + ", c=" + c + ", m=" + m + ", s=" + s + ", d=" + d + ", dead=" + dead + "]";
      }

   }
   
   static class Place{
      int r;
      int c;
      public Place(int r, int c) {
         this.r = r;
         this.c = c;
      }
      @Override
      public String toString() {
         return "Place [r=" + r + ", c=" + c + "]";
      }

      @Override
      public int hashCode()
      {
       return Integer.toString(r).hashCode()+Integer.toString(c).hashCode();
      }
      
      @Override
      public boolean equals(Object o)
      {
		   if(o==null)
			   return false;
		   
		   Place temp=(Place) o;
		   
		   if(temp.r==this.r && temp.c==this.c)
			   return true;
		   else
			   return false;
      }
   }
   
   
   private static int [] dr = new int[]{-1, -1, 0, 1, 1, 1, 0, -1};
   private static int [] dc = new int[]{0, 1, 1, 1, 0, -1, -1, -1};

   public static void main(String[] args) throws IOException {
      BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
      StringTokenizer st = new StringTokenizer(br.readLine());
      
      N = Integer.parseInt(st.nextToken()); // N * N 배열
      M = Integer.parseInt(st.nextToken());
      K = Integer.parseInt(st.nextToken()); // 명령 횟수! == 이동 횟수1

      List<Fireball> fireballs = new ArrayList<>();
      
      for (int i = 0; i < M; i++) {
         st = new StringTokenizer(br.readLine());
         
         int r = Integer.parseInt(st.nextToken()) - 1;
         int c = Integer.parseInt(st.nextToken()) - 1;
         int m = Integer.parseInt(st.nextToken()); 
         int s = Integer.parseInt(st.nextToken()); 
         int d = Integer.parseInt(st.nextToken());
         
         fireballs.add(new Fireball(r,c,m,s,d,false));
      }
      
      for (int i = 0; i < K; i++) {
          Set<Place> set = new HashSet<>();
          Map<Place, List<Integer>> map = new HashMap<>();
          
          List<Fireball> newFireballs = new ArrayList<>();
         // 1. 모든 파이어볼이 자신의 방향 di로 속력 si칸 만큼 이동한다. 이동하는 중에는 같은 칸에 여러 개의 파이어볼이 있을 수도 있다.
//          System.out.println("===============");
         for (int j = 0; j < fireballs.size(); j++) {
            Fireball cur = fireballs.get(j);
            
            if(cur.isDead()) continue;
            
            if(cur.m == 0) {
            	cur.setDead(true);
            }
            
            int nr = cur.r + (dr[cur.d] * cur.s);
            int nc = cur.c + (dc[cur.d] * cur.s);
            
            if(nr >= N) {
               nr = nr % N;
            } else if(nr < 0) {
            	nr = nr % N + N;
            }
         
            if(nc >= N) {
            	nc = nc % N;
            } else if(nc < 0) {
            	nc = nc % N + N;
            }
            cur.setR(nr);
            cur.setC(nc);
            
      
            Place pl = new Place(cur.r, cur.c);

            if(map.get(pl) != null) {
                List<Integer> al = map.get(pl);
                al.add(j);
                map.put(pl, al);
            } else {
                List<Integer> al = new ArrayList<>();
                al.add(j);
                map.put(pl, al);
            }
         }
         
         // 2. 이동이 모두 끝난 뒤, 2개 이상의 파이어볼이 있는 칸에서는 다음과 같은 일이 일어난다.
         
            Iterator<Place> keys = map.keySet().iterator();
            while (keys.hasNext()){
                Place key = keys.next();
                List<Integer> list = map.get(key);
                if(list.size() <= 1) {
                    int idx = list.get(0);
	                Fireball temp = fireballs.get(idx);
	                newFireballs.add(new Fireball(key.r, key.c, temp.m, temp.s, temp.d, false));
                   continue;
                } else {
//                    System.out.println(key.toString());
//                    System.out.println(list.toString());
                   int sumM = 0;
                   int sumS = 0;
                   
                   boolean two = false;
                   boolean notTwo = false;
                   for (int j = 0; j < list.size(); j++) {
                      int idx = list.get(j);
	                  Fireball temp = fireballs.get(idx);
	                  fireballs.get(idx).setDead(true);

	                  sumM += temp.m;
	                  sumS += temp.s;
	                  
	                  if(temp.d % 2 == 0) {
	                     two = true;
	                  } else {
	                     notTwo = true;
	                  }
                   }
                   
                   // 그냥 (int) (sumM / 5) 해도 틀림
                   int newM = (int) Math.floor(sumM / 5);
                   int newS = (int) Math.floor(sumS / list.size());
                   
                   if(newM == 0) continue;
                   
                   if(two == true && notTwo == true) { // 모두 홀수도 모두 짝수도 아니다.
                	   newFireballs.add(new Fireball(key.r, key.c, newM, newS, 1, false));
                	   newFireballs.add(new Fireball(key.r, key.c, newM, newS, 3, false));
                	   newFireballs.add(new Fireball(key.r, key.c, newM, newS, 5, false));
                	   newFireballs.add(new Fireball(key.r, key.c, newM, newS, 7, false));
                   } else {
                	   newFireballs.add(new Fireball(key.r, key.c, newM, newS, 0, false));
                	   newFireballs.add(new Fireball(key.r, key.c, newM, newS, 2, false));
                	   newFireballs.add(new Fireball(key.r, key.c, newM, newS, 4, false));
                	   newFireballs.add(new Fireball(key.r, key.c, newM, newS, 6, false));
                   }
                }
            }
            
            fireballs = newFireballs;
      }
      int answer = 0;
      for (int i = 0; i < fireballs.size(); i++) {
         Fireball cur = fireballs.get(i);
         if(cur.isDead()) continue;
         System.out.println(cur.toString());
         answer += cur.m;
      }
      
      System.out.println(answer);
   }

}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

예제는 다 맞는데 왜 틀렸습니다!가 나오는지 모르겠습니다!!

seulshine 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말