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

mini98님의 프로필 이미지
mini98

작성한 질문수

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

4-F

재귀로 탐색 시 종료조건에 대한 문의입니다.

작성

·

402

1

안녕하세요 큰돌님! 풀이를 보기 전에 제 식대로 일단 풀어봤는데요

통과한 코드

https://www.acmicpc.net/source/56817951

실패한 코드

https://www.acmicpc.net/source/56817956

두 풀이 모두 8 line의 dfs의 종료 조건을 제외하고 동일한 코드입니다.

결과는 v==26에서는 실패하고 v==27에서 성공합니다.

dfs(int v, int on)의 v는 v번째 알파벳을 킬것인지 끌것인지 여부인데요,

제 생각에는 0~25까지 a부터 z를 담당하기 때문에 v==26 이 되면, 0~25까지 킬 것인지 끌것인지를 모두 확인했기 때문에, 그 때 종료를 시켜야 할 것 같은데

v ==26에서는 풀이가 실패하고 v==27일 때 풀이가 통과해서 제 예상과는 다르게 동작합니다.

이유를 알 수 있을까요?

 

좋은 강의 감사합니다.

답변 1

0

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

안녕하세요 98님 ㅎㅎ

잘 짜셨고 26으로 해도 맞게 나올 수 있습니다. 다만 우선순위 문제인 거 같아요.

 

이걸 잠깐 볼까요?

		if (!(mask & (1<<v))) {
			mask |= (1<<v);
			dfs(v+1, on+1);

자 지금 보면 해당 v번째를 추가하고 ~~ dfs v + 1을 통해서 v + 1에서 종료를 시키고 있죠?

0부터 시작해볼께요.

 

0, 0

>>

1, 1을 하면서 0을 집어넣었다. 라는 뜻으로 되는 거구요.

쭉 이렇게 가다보면

25, 5 가 되고 이는 24를 집어넣었다. 가 되고

그 다음.

26, 6 이는 25를 집어넣었다. 가 됩니다.

근데 여기서 26일 때 종료를 시키면

해당 26을 집어넣었을 때의 이로직보다.

if (on==k) {
		int cnt = 0;
		for (int i=0; i<n; i++) {
			int word = words[i];
			if ((word&mask) == word) cnt++;
		}
		ret = max(ret, cnt);
		return;
	}

다음 로직이 먼저 실행이 되겠죠?

	if (v==26) return;

그러면 26을 집어넣었을 때의 해당 카운팅 로직이 실행될까요?

안되겠죠?

그럼 어떻게 코드를 수정해야할까요?

 

다음과 같이 우선순위를 고치시면 됩니다.

#include<bits/stdc++.h>
using namespace std;

int n,k,words[51], mask, ret;
string s, need = "acint";

void dfs(int v, int on) {  
    if (on==k) {
		int cnt = 0;
		for (int i=0; i<n; i++) {
			int word = words[i];
			if ((word&mask) == word) cnt++;
		}
		ret = max(ret, cnt);
		return;
	} 
	if (v==26) return; 
    if (!(mask & (1<<v))) {
        mask |= (1<<v);
        dfs(v+1, on+1);
        mask &= (~(1<<v));
    }
    dfs(v+1, on); 
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	
	cin >> n >> k;
	for (int i=0; i<n; i++) {
		cin >> s;
		s= s.substr(4, s.size()-8);
		for (char c : s) {
			words[i] |= (1 << (c-'a'));
		}
	}
	if (k < 5) {
		cout << "0" << '\n';
		return 0;
	}
	for (char c : need) {
		mask |= (1 << (c-'a'));
	}
	k -=5;
	dfs(0,0);
	cout << ret << '\n';
}

감사합니다.

mini98님의 프로필 이미지
mini98
질문자

답변주셔서 감사합니다! 덕분에 열심히 공부할 수 있는 것 같습니다

mini98님의 프로필 이미지
mini98

작성한 질문수

질문하기