본문 바로가기
728x90

분류 전체보기423

[python] 백준 1748 수 이어 쓰기 1 https://www.acmicpc.net/problem/1748 1748번: 수 이어 쓰기 1 첫째 줄에 N(1 ≤ N ≤ 100,000,000)이 주어진다. www.acmicpc.net 문제 해결 입력 값 n이 몇자리 수인지 구한다. n이 x 자리 수이면 x-1 자리 수까지 임의의 i자리 수는 각각 9$\times 10^{i-1}$ 개가 있다. 예로들면 400인 3자리수를 보면 한자리 수는 1~9 9개가 있고 두자리수는 10~99 90개가 있다. 따라서 i자리 수 모두를 이어 붙이면 i $\times 9 \times 10^{i-1}$ 자리가 된다. 나머지 x자리 수 개수 (n- ($10^{x-1}$-1)) $\times$ x 를 붙이면 n까지 수를 이어스 쓴 자릿수가 된다. CODE import sy.. 2023. 1. 3.
[python] 백준 14500 테트로미노 https://www.acmicpc.net/problem/14500 14500번: 테트로미노 폴리오미노란 크기가 1×1인 정사각형을 여러 개 이어서 붙인 도형이며, 다음과 같은 조건을 만족해야 한다. 정사각형은 서로 겹치면 안 된다. 도형은 모두 연결되어 있어야 한다. 정사각형의 변 www.acmicpc.net 문제 해결 사실 너무 어려웠다. 한 점을 잡고 이동을 해야할 것 같은데 어떻게 해야할 지 몰랐다. 이 모양을 제외 하면 모두 한 붓 그리기, 즉 다시 되돌아가지 않고 이동하면서 그릴 수 있다는 특징이 있다. 이 모형도 가운데를 중심으로 동,서,남,북 으로 이동 후 하나를 자른다는 생각을 할 수 있다. 자르는 것은 최소의 값을 가진 지역이라 할 수 있다. 이동하는 방법을 move = [ (1,0),.. 2023. 1. 2.
[PyTorch] masked_fill() 함수 사용 예시 파이토치에서 masked_fill() 함수는 파이토치 tensor 의 특정 값을 다른 값으로 바꾸고자 할 때 사용된다. document 가 좀 설명이 부실한 것 같아서 예시로 하나 사용해보면 import torch import torch.nn.functional as F example = torch.FloatTensor([[1, 2, 3], [4, 5, 6]]) # 특정 행렬 mask = torch.FloatTensor([[0, 1, 1], [1, 0, 1]]) # 바꾸고자 하는 행렬의 원소를 0 고정하고자 하는 행렬의 원소를 1로 둔다. result = example.masked_fill(mask == 0, 5.5) # 0을 5.5로 변환 print(result) 2023. 1. 2.
[PyTorch] torch.transpose()와 numpy().transpose() 차이 파이토치의 transpose는 N $\times$ M 만 transpose 할 수 있다는 단점이 있다. 그러면 N $\times$ M $\times$ K는 어떻게 나타낼 수 있을까? => numpy로 바꾼 후 transpose() 하면 된다. import torch import numpy as np x = torch.rand(3,4,2) print(x.numpy().transpose((1,2,0))) 2023. 1. 1.
728x90