Розглянемо, як поширюється вогонь на нескінченній трикутній решітці:
Початкова вершина. У момент часу горить тільки початкова вершина. Це дає першу одиницю в загальному числі палаючих вершин.
Перший рівень (через секунду). Вершини, які безпосередньо сусідять із початковою, загоряться через секунду. У трикутній решітці кожна вершина має сусідів. Отже, перший рівень містить вершин.
Другий рівень (через секунди). На наступну секунду загоряться вершини, які сусідять із уже палаючими вершинами першого рівня, але ще не горять. Другий рівень містить вершин.
Можна помітити закономірність: кожен новий рівень додає на вершин більше, ніж попередній.
Перший рівень: вершин
Другий рівень: вершин
Третій рівень: вершин
... і так далі.
Маємо арифметичну прогресію з першим членом a_1 = 6 і різницею d = 6.
Через секунд кількість палаючих вершин буде дорівнювати
Використовуючи формулу суми арифметичної прогресії \frac{2a_1+d(n-1)}{2} n, отримаємо що
Реалізація алгоритму
Читаємо вхідне значення .
scanf("%lld",&n);Обчислюємо і виводимо відповідь.
res = 1 + 3 * n * (n + 1);
printf("%lld\n",res);