题解：P11871 张灯结彩

yyycj · 2025-03-09 19:36:58 · 题解

主要思路

其实题目中写的已经很清楚了，对于 i \in [1,n)，把每个彩灯与其下一层最相邻的三个彩灯连上电线，该层向下连接的电线数就是 (2i-1) \times 3；对于 i=n，把两两相邻的彩灯连上电线，该层的电线数就是 2i-1-1。

AC Code

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;

typedef long long ll;
typedef long double db;
// ----------------------------

// ----------------------------

// ----------------------------

int main() {
    int t; cin >> t;
    // ----------------------------
    int n, ans;
    while (t--) {
        cin >> n;
        ans = 2 * n - 2;
        for (int i = 1; i < n; i++) ans += (2 * i - 1) * 3;
    cout << ans << '\n';
    }
    return 0;
}