백준 1439번 - 뒤집기 / Go

문제: https://www.acmicpc.net/problem/1439

1439번: 뒤집기

문제 설명

문제

다솜이는 0과 1로만 이루어진 문자열 S를 가지고 있다. 다솜이는 이 문자열 S에 있는 모든 숫자를 전부 같게 만들려고 한다. 다솜이가 할 수 있는 행동은 S에서 연속된 하나 이상의 숫자를 잡고 모두 뒤집는 것이다. 뒤집는 것은 1을 0으로, 0을 1로 바꾸는 것을 의미한다.

예를 들어 S=0001100 일 때,

1. 전체를 뒤집으면 1110011이 된다.
2. 4번째 문자부터 5번째 문자까지 뒤집으면 1111111이 되어서 2번 만에 모두 같은 숫자로 만들 수 있다.

하지만, 처음부터 4번째 문자부터 5번째 문자까지 문자를 뒤집으면 한 번에 0000000이 되어서 1번 만에 모두 같은 숫자로 만들 수 있다.

문자열 S가 주어졌을 때, 다솜이가 해야하는 행동의 최소 횟수를 출력하시오.

입력

첫째 줄에 문자열 S가 주어진다. S의 길이는 100만보다 작다.

출력

첫째 줄에 다솜이가 해야하는 행동의 최소 횟수를 출력한다.

예제 입력 1 복사

0001100

예제 출력 1 복사

1

예제 입력 2 복사

11111

예제 출력 2 복사

0

예제 입력 3 복사

00000001

예제 출력 3 복사

1

예제 입력 4 복사

11001100110011000001

예제 출력 4 복사

4

예제 입력 5 복사

11101101

예제 출력 5 복사

2
​

출처

• 문제를 번역한 사람: baekjoon
• 어색한 표현을 찾은 사람: jh05013

알고리즘 분류

• 문자열
• 그리디 알고리즘

 

---

정답

package main

import (
	"bufio"
	"fmt"
	"math"
	"os"
	"regexp"
	"strings"
	"testing"
)

func main() {
	rd := bufio.NewReader(os.Stdin)
	wr := bufio.NewWriter(os.Stdout)

	word := scan1439(rd)

	_, _ = wr.WriteString(solution1439(word))
	_ = wr.Flush()
}

func solution1439(word string) (result string) {
	rgxZero, _ := regexp.Compile("0+")
	zeros := rgxZero.FindAllString(word, -1)
	rgxOne, _ := regexp.Compile("1+")
	ones := rgxOne.FindAllString(word, -1)

	x := float64(len(zeros))
	y := float64(len(ones))
	result = fmt.Sprintf("%.0f", math.Min(x, y))
	return
}

func scan1439(rd *bufio.Reader) string {
	str, _ := rd.ReadString('\n') // 여기서 text는 마지막에 줄바꿈 문자를 포함하므로
	str = strings.TrimSpace(str)  // 줄바꿈 문자를 제거해야 함
	return str
}

func Benchmark1439(b *testing.B) {
	for i := 0; i < b.N; i++ {
		solution1439("101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101")
	}
}

func Test_solution1439(t *testing.T) {
	type args struct {
		word string
	}
	tests := []struct {
		name       string
		args       args
		wantResult string
	}{
		{name: "", args: args{word: "0001100"}, wantResult: "1"},
		{name: "", args: args{word: "11111"}, wantResult: "0"},
		{name: "", args: args{word: "00000001"}, wantResult: "1"},
		{name: "", args: args{word: "11001100110011000001"}, wantResult: "4"},
		{name: "", args: args{word: "11101101"}, wantResult: "2"},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if gotResult := solution1439(tt.args.word); gotResult != tt.wantResult {
				t.Errorf("solution1439() = %v, want %v", gotResult, tt.wantResult)
			}
		})
	}
}

 

---

풀이

 정규식을 공부하면서 정규식으로 문제를 풀어본 첫 문제이다.

 

1개 이상의 0, 1개 이상의 1로 이루어진 단어의 개수를 찾은 다음 둘 중 최솟값을 리턴하면 되는 문제이다.