글 읽기 - 제가 이 문제 잘못 이했나요?

akswnd98 4년 전 0

저는 이 문제가

su = 0

for all i, j:

su += (a(i, j) == 1 ? 1 : 0) * (i + 1) * (j + 1) * (N - i) * (M - j)

로 푸는 건줄 알았는데 아닌가요?

진짜 알 수 없는 이유로 계속 틀립니다.

import java.util.Scanner;
import java.util.Arrays;
import java.util.ArrayList;
import java.util.LinkedList;

class Main {
  private static class Pair {
    public int first, second;

    public Pair(int first, int second) {
      this.first = first;
      this.second = second;
    }
  }

  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    int N = sc.nextInt();
    int M = sc.nextInt();
    int K = sc.nextInt();
    int[][] a = new int[N][M];
    int i, j, k;
    int su = 0;
    Pair[] link = new Pair[K];
    for(i = 0; i < K; i++) {
      link[i] = new Pair(-1, -1);
    }

    for(i = 0; i < N; i++) {
      for(j = 0; j < M; j++) {
        a[i][j] = sc.nextInt();
        if(a[i][j] == 1) {
          su += (i + 1) * (j + 1) * (N - i) * (M - j);
          su %= K;
        }
      }
    }
    link[su].first = su;
    link[su].second = -1;
    for(i = 0; i < N; i++) {
      for(j = 0; j < M; j++) {
        if(a[i][j] < 0) {
          for(k = 0; k < K; k++) {
            if(link[k].first >= 0 && link[(k + (i + 1) * (j + 1) * (N - i) * (M - j)) % K].first < 0) {
              link[(k + (i + 1) * (j + 1) * (N - i) * (M - j)) % K].first = k;
              link[(k + (i + 1) * (j + 1) * (N - i) * (M - j)) % K].second = i * M + j;
            }
          }
        }
      }
    }
    if(link[0].first < 0) {
      System.out.println(-1);
      return;
    }
    i = 0;
    while(link[i].second >= 0) {
      a[link[i].second / M][link[i].second % M] = 1;
      i = link[i].first;
    }
    for(i = 0; i < N; i++) {
      for(j = 0; j < M; j++) {
        if(a[i][j] < 0) {
          a[i][j] = 0;
        }
      }
    }

    System.out.println(1);
    for(i = 0; i < N; i++) {
      for(j = 0; j < M; j++) {
        System.out.printf("%d ", a[i][j]);
      }
      System.out.println();
    }
  }
}

akswnd98 4년 전 0

코드가 이상했습니다.

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

문제

문제

출처

ICPC

제가 이 문제 잘못 이했나요?

akswnd98 4년 전 0

akswnd98 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말