[Swift] 2019 KAKAO BLIND RECRUITMENT 매칭 점수

Problem

 

코딩테스트 연습 - 매칭 점수

---

Solution

 

1. 페이지에 필요한 정보를 담는 객체를 만들어준다.

 

몇번째 인덱스인지,자신의 url,총 점수,외부 링크를 프로퍼티로 설정해줍니다.

 

struct Page {
    var index:Int, url:String,score:Double,externalLinks:[String]
}

 

2. "<" 기준으로 page의 문자열을 자른다.

 

let tags = page.split(separator: "<")

 

3. 나눠진 태그별로 정보를 알아낸다.

 

만약 앞 글자가 meta p(meta property)일 경우 자신의 url을 뜻하므로 url을 설정해줍니다. (소스 코드에서 getWebpageURL를 참고해주세요!)

 

/body일 경우 body가 끝난 것이므로 점수를 세팅해줍니다.

 

a href일 경우 외부 링크를 뜻하므로 외부링크를 세팅해줍니다.

 

body일 경우 score에 +1을 해주어 0으로 만들어줍니다.

 

그 외 일 경우에 body안에 word를 찾아내어 점수에 추가해줍니다. (소스 코드에서 findWordInBody를 참고해주세요!)

 

var pageInfos:[Page] = []
    for (i,page) in pages.enumerated() {
        var pageInfo = Page(index:i,url: "", score: 0, externalLinks: [])
        let tags = page.split(separator: "<")
        var score:Double = -1
        
        for tag in tags {
            if String(tag.prefix(6)) == "meta p" {
                pageInfo.url = getWebpageURL(tag: String(tag))
            }
            if String(tag.prefix(5)) == "/body" {
                pageInfo.score = score
                break
            }
            if String(tag.prefix(6)) == "a href" {
                let url = tag.split(separator: "\"")[1]
                pageInfo.externalLinks.append(String(url))
            }
            if String(tag.prefix(4)) == "body" {
                score += 1
            }
            if score > -1 {
                findWordInBody(tag: String(tag), word: word.lowercased(), score: &score)
            }
        }
        pageInfos.append(pageInfo)
    }

 

3. 매칭 점수가 가장 높은 인덱스를 찾아낸다.

 

func getMaxMatchingScoreIndex(pageInfos:inout [Page]) -> Int {
    for pageInfo in pageInfos {
        for link in pageInfo.externalLinks {
            if let index = pageInfos.firstIndex(where: {$0.url == link}) {
                pageInfos[index].score += pageInfo.score/Double(pageInfo.externalLinks.count)
            }
        }
    }
    return pageInfos.sorted{($0.score) > ($1.score)}.first?.index ?? 0
}

 

4. 매칭 점수가 가장 높은 인덱스를 반환한다.

 

return getMaxMatchingScoreIndex(pageInfos: &pageInfos)

---

Source Code

 

	import Foundation
	struct Page {
	var index:Int, url:String,score:Double,externalLinks:[String]
	}
	func solution(_ word:String, _ pages:[String]) -> Int {
	var pageInfos:[Page] = []
	for (i,page) in pages.enumerated() {
	var pageInfo = Page(index:i,url: "", score: 0, externalLinks: [])
	let tags = page.split(separator: "<")
	var score:Double = -1
	for tag in tags {
	if String(tag.prefix(6)) == "meta p" {
	pageInfo.url = getWebpageURL(tag: String(tag))
	}
	if String(tag.prefix(5)) == "/body" {
	pageInfo.score = score
	break
	}
	if String(tag.prefix(6)) == "a href" {
	let url = tag.split(separator: "\"")[1]
	pageInfo.externalLinks.append(String(url))
	}
	if String(tag.prefix(4)) == "body" {
	score += 1
	}
	if score > -1 {
	findWordInBody(tag: String(tag), word: word.lowercased(), score: &score)
	}
	}
	pageInfos.append(pageInfo)
	}
	return getMaxMatchingScoreIndex(pageInfos: &pageInfos)
	}
	func getWebpageURL(tag:String) -> String {
	let content = tag.split(separator: " ").last!
	return String(content.split(separator: "\"")[1])
	}
	func findWordInBody(tag:String,word:String,score:inout Double) {
	let contains = tag.split(separator: " ").filter{$0.lowercased().contains(word)}
	for contain in contains {
	let new = contain.map{$0.isLetter ? String($0.lowercased()) : " "}.joined().split(separator: " ")
	score += Double(new.filter{$0 == word}.count)
	}
	}
	func getMaxMatchingScoreIndex(pageInfos:inout [Page]) -> Int {
	for pageInfo in pageInfos {
	for link in pageInfo.externalLinks {
	if let index = pageInfos.firstIndex(where: {$0.url == link}) {
	pageInfos[index].score += pageInfo.score/Double(pageInfo.externalLinks.count)
	}
	}
	}
	return pageInfos.sorted{($0.score) > ($1.score)}.first?.index ?? 0
	}

view raw 매칭 점수.swift hosted with ❤ by GitHub

---

P.S

 

테스트 케이스 8번만 틀려서 찾아봤더니 총 점수를 Double로 해야하는 문제였다...