[C++]백준 2481 - 데스노트

2281번: 데스노트

풀이

---

DP를 이용해 푸는 문제다.

한줄 넘기고 이름을 쓰는 경우와 넘기지 않고 이어서 이름을 쓰는 경우 두가지로 나누어서 풀면된다.

 

코드

---

#include <iostream>
#include <memory.h>
#include <math.h>
using namespace std;
int m,n;
int name[1001];
int dp[1001][1001] = {0};

int DeathNote(int nameIndex, int length) {
	if (nameIndex >= n) {
		return 0 ;
	}
	if (dp[nameIndex][length] != -1)
		return dp[nameIndex][length];
	int pow = (m - length + 1) * (m - length + 1);
	dp[nameIndex][length] = pow + DeathNote(nameIndex + 1, name[nameIndex] + 1);
	if (name[nameIndex] + length <= m) {
		dp[nameIndex][length] = min(dp[nameIndex][length], DeathNote(nameIndex + 1, length + name[nameIndex] + 1));
	}
	return dp[nameIndex][length];
}

int main() {
	cin >> n >> m;
	for (int i = 0; i < n; i++) {
		cin >> name[i];
	}
	memset(dp, -1, sizeof(dp));
	cout << DeathNote(0, 0);
	return 0;
}