글 읽기 - 살려주세요 ㅜㅜ 99%에서 틀립니다...

yubhop 4년 전 1

99%에서 틀린다고 나옵니다.

보통 99%면 아주 특이한 케이스라는데 감을 못잡겠네요...

답변 주시면 감사하겠습니다!

#include <iostream>
#include <stdio.h>
#include <algorithm>

long long arr[1000001] = { -1, };
long long sum[1000001] = { 0, };
long long re[1000001] = { 1001, };

int main() {
	int N, M;
	long long cnt = 0;
	long long k = 1;
	scanf("%d %d", &N, &M);
	for (int i = 1; i <= N; i++) {
			scanf("%lld", &arr[i]);
			sum[i] = sum[i - 1] + arr[i];
			re[i] = sum[i] % M;
	}		
	std::sort(re+1, re + N + 1);
	if (N==1) {
		if (M%arr[1]==0) {
			printf("%d", 1);
			return 0;
		}
	}
	for (int i = 1 ; i<= N ; i++) {
		if (re[i] == re[i+1]) {
			k++;
		}
		else {
			if (re[i]==0) {
				if (k==1) {
					cnt++;
				}
				else {
					cnt+=k*(k-1)/2+k;
				}
			}
			else if (k>=2) {
				cnt+=(k-1)*(k)/2;
			}
			k = 1;
		}
	}
	printf("%lld", cnt);
	return 0;
}

clrmt 4년 전 1

2 2

하면 0이 나옵니다.

cs71107 4년 전 2

반례)

5 3

0 0 0 0 0

위와 같은 테스트 케이슬을 넣어보시면, 올바른 답은 15이지만, 이 코드에서는 0을 출력합니다.

re[i]==re[i+1]을 실행할 때, re[n+1]이 0이기 때문에, 미리 절대로 M으로 나눈 나머지가 될 수 없는 값으로 바꿔놔야 합니다.

그리고, N=1일 때, M%arr[1]이 아니라 arr[1]%M으로 해주세요.

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

문제

문제

출처

ICPC

살려주세요 ㅜㅜ 99%에서 틀립니다...

yubhop 4년 전 1

clrmt 4년 전 1

cs71107 4년 전 2

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말