Иногда встречаются задачи с дву/многомерными массивами, в которых есть несколько очевидных случаев возможного выхода за пределы массива. Они просты и понятны, но для их обработки порой требуется много "лишнего" кода (однажды автор написал 16 if-ов!). Я придумал простой способ, который поможет избежать этого. Возможно, это покажется вам очевидным, но лично мне так не казалось еще вчера. Думаю, я не один такой.
Пример задачи (как можно более простой, взята отсюда http://informatics.mccme.ru/mod/statements/view3.php?chapterid=946 ):
Дана прямоугольная доска N × N (N <= 20). Конь стоит в верхнем левом углу доски. Выведите количество способов добраться конём до правого нижнего угла доски, если конь может ходить только так:
Вот обычное решение этой задачи динамическим программированием:
#include <iostream>
int n, dp[20][20];
int main() {
std::cin >> n;
dp[0][0] = 1;
for (int i = 1; i < n; i++) {
for (int j = 1; j < n; j++) {
if (i-2 >= 0 && j-1 >= 0)
dp[i][j] += dp[i-2][j-1];
if (i-1 >= 0 && j-2 >= 0)
dp[i][j] += dp[i-1][j-2];
}
}
std::cout << dp[n-1][n-1] << '\n';
}
Казалось бы, ничего сложного. Но тем не менее его можно упростить:
#include <iostream>
int n, dp[20][20];
int f(int i, int j) {
return (i < 0 || j < 0) ? 0 : dp[i][j];
}
int main() {
std::cin >> n;
dp[0][0] = 1;
for (int i = 1; i < n; i++)
for (int j = 1; j < n; j++)
dp[i][j] = f(i-2, j-1) + f(i-1, j-2);
std::cout << dp[n-1][n-1] << '\n';
}
В данном примере не видно большой разницы, но в некоторых случаях это может действительно сократить код, тем самым уменшив возможность ошибки. Это способ хорош еще тем, что если вдруг надо будет возвращать не 0 в случае выхода за пределы, а, например -inf, то это легко сделать.
Кстати, функцию f можно сделать более гибкой, передав ей массив/вектор в качестве аргумента, например: int f(vector<int>& v, int i, int j) {
, а можно даже использовать C++14 и написать так: int f(auto& v, int i, int j) {
.
Твой код не обрабатывает случай когда индекс больше либо равен размеру массива
Обычно я делаю так
Я бы так делал:
Может,
dp[1][1] = 1;
?Да, точно, спасибо.