[프로그래머스] 모음 사전 c++

semi·2021년 9월 1일
0

coding test

목록 보기
3/57

https://programmers.co.kr/learn/courses/30/lessons/84512

#include <iostream>
#include <string>
#include <vector>
#include <map>
using namespace std;

int solution(string word)
{
	int answer = 0;
	int mul[5] = { 1, 5, 25, 125, 625 };
	map<char, int> m = { {'A', 0}, {'E', 1}, {'I', 2}, {'O', 3}, {'U', 4} };
	for (int i = 0; i < word.size(); i++)
	{
		answer += 1;
		for (int j = 0; j < m[word[i]]; j++)
		{
			for (int k = 4; k >= i; k--)
			{
				answer += mul[4 - k];
			}
		}


	}
	return answer;
}

int main(void)
{
	string word1 = "AAAAE", word2 = "AAAE", word3 = "I", word4 = "EIO";
	cout << solution(word4);
	return 0;
}

0개의 댓글