강의

멘토링

로드맵

Inflearn brand logo image

인프런 커뮤니티 질문&답변

Jeongmin Kim님의 프로필 이미지
Jeongmin Kim

작성한 질문수

10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트

7-T

7-T 5557번 1학년 문제 질문입니다.

해결된 질문

작성

·

74

0

일단 제가 틀린 코드인데,

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int n, a[102];
ll dp[102][102];
// sum은 음수가 되면 안됨. 20을 넘어서도 안됨.
ll go(int sum, int cnt) {
	if (sum < 0 || sum > 20) return 0;
	if (cnt == n - 1) {
		if (sum == a[n]) return 1;
		else return 0;
	}
	ll& ret = dp[sum][cnt];
	if (ret != -1) return ret;
	ret = 0;
	ret += go(sum + a[cnt + 1], cnt + 1);
	ret += go(sum - a[cnt + 1], cnt + 1);
	return ret;
}
int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> n;
	memset(dp, -1, sizeof(dp));
	for (int i = 1; i <= n; ++i) {
		cin >> a[i];
	}
	cout << go(0, 0);
}

물론 이 방법이 불필요한 과정이 들어있는 것은 맞지만, 예제도 다 통과했고 결국 같은 과정을 거친다고 생각했는데, 오답이라고 나옵니다.

그래서 a[0]부터 접근하는 코드로 변경했을 때는 정답이라고 나오는데 어떤 차이점인지 잘 모르겠습니다.
https://inf.run/hL4B9

답변 2

1

큰돌님의 프로필 이미지
큰돌
지식공유자

.

0

Jeongmin Kim님의 프로필 이미지
Jeongmin Kim
질문자

처음에 0이 나오는 경우의 수를 생각하지 못했네요.. 제 방식으로 하면 처음에 0으로 시작하는 경우의 수가 두 가지 추가 되네요.

큰돌님의 프로필 이미지
큰돌
지식공유자

딱 말씀드리려는데 알아차리셨네요 ㅎㅎ 넴넴

	ret += go(sum + a[cnt + 1], cnt + 1);
	ret += go(sum - a[cnt + 1], cnt + 1);
	return ret;

코드보시면a[1]부터 추가되서.. 해당 부분이 사라지게 됩니다.

 

감사합니다.

Jeongmin Kim님의 프로필 이미지
Jeongmin Kim

작성한 질문수

질문하기