BAEKJOON: while 문

10951: A+B - 4

문제:
두 정수 A와 B를 입력받은 다음, A+B를 출력하는 프로그램을 작성하시오.

입력:
입력은 여러 개의 테스트 케이스로 이루어져 있다.

각 테스트 케이스는 한 줄로 이루어져 있으며, 각 줄에 A와 B가 주어진다. (0 < A, B < 10)

출력:
각 테스트 케이스마다 A+B를 출력한다.

package main

import (
	"bufio"
	"fmt"
	"os"
)

var reader *bufio.Reader = bufio.NewReader(os.Stdin)
var writer *bufio.Writer = bufio.NewWriter(os.Stdout)

func main() {
	defer writer.Flush()
	for {
		var A, B int
		val, _ := fmt.Fscanln(reader, &A, &B)
		if val != 2 {
			break
		}
		fmt.Fprintln(writer, A + B)
	}
}

1110: 더하기 사이클

문제:
0보다 크거나 같고, 99보다 작거나 같은 정수가 주어질 때 다음과 같은 연산을 할 수 있다. 먼저 주어진 수가 10보다 작다면 앞에 0을 붙여 두 자리 수로 만들고, 각 자리의 숫자를 더한다. 그 다음, 주어진 수의 가장 오른쪽 자리 수와 앞에서 구한 합의 가장 오른쪽 자리 수를 이어 붙이면 새로운 수를 만들 수 있다. 다음 예를 보자.

26부터 시작한다. 2+6 = 8이다. 새로운 수는 68이다. 6+8 = 14이다. 새로운 수는 84이다. 8+4 = 12이다. 새로운 수는 42이다. 4+2 = 6이다. 새로운 수는 26이다.

위의 예는 4번만에 원래 수로 돌아올 수 있다. 따라서 26의 사이클의 길이는 4이다.

N이 주어졌을 때, N의 사이클의 길이를 구하는 프로그램을 작성하시오.

입력:
첫째 줄에 N이 주어진다. N은 0보다 크거나 같고, 99보다 작거나 같은 정수이다.

출력:
첫째 줄에 N의 사이클 길이를 출력한다.

package main

import (
	"bufio"
	"fmt"
	"os"
)

var reader *bufio.Reader = bufio.NewReader(os.Stdin)
var writer *bufio.Writer = bufio.NewWriter(os.Stdout)

func main() {
	defer writer.Flush()
	var N, res, cycle int
	fmt.Fscan(reader, &N)
	res = N
	for {
		x, y := getXY(res)
		res = getNum(x, y)
		cycle++
		if res == N {
			fmt.Fprintln(writer, cycle)
			break
		}
	}
}

func getNum(x int, y int) int {
	z := (x + y) % 10
	num := y * 10 + z
	return num
}

func getXY(num int) (int, int) {
	x := num / 10 % 10
	y := num % 10
	return x, y
}