Gym - 101635J - Frosting on the Cake

Gym - 101635J - Frosting on the Cake

TLE代码

画面：我天真地以为不能再优化了，自信地交了上去。

#include <bits/stdc++.h>
using namespace  std;
using ll = long long;
int main() {
    ios::sync_with_stdio(false);

    int n; cin >> n;
    vector<ll> w(n + 1, 0), l(n + 1, 0);
    for (int i = 1 ;i <= n; i++) cin >> w[i];
    for (int i = 1 ;i <= n; i++) cin >> l[i];
    vector<ll> ans(3, 0);
    // 啊，昨天刚写了用上三角，这就用上了，然后WA了hhhhh
    for (int i = 1; i <= n; i++) {
        ans[(i + i) % 3] += w[i] * l[i];
        for (int j = i + 1; j <= n; j++) {
            ans[(i + j) % 3] += ((w[i] * l[j]) + (w[j] * l[i]));
        }
    }
    for (auto it : ans) cout << it << ' ';
    return 0;
}

AC代码

看起来离散的Blocks，却有着简单直接的联系。啊，这nm一眼就能看出来他是第一个3*3的复制粘贴啊啊啊啊。
一直在斜着看它的副对角线忽视这一点的我也是很强了。

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
int main() {
    int n; cin >> n;
    vector<ll> a(n + 1, 0), b(n + 1, 0);
    vector<ll> ans1(3), ans2(3);
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        ans1[i % 3] += a[i];
    }
    for (int i = 1; i <= n; i++) {
        cin >> b[i];
        ans2[i % 3] += b[i];
    }
    ll w = 0, y = 0, p = 0;
    // 这里照着图中的第一个3*3来写就会十分简单。
    w = ans1[1] * ans2[1] + ans1[2] * ans2[0] + ans1[0] * ans2[2];
    y = ans1[2] * ans2[1] + ans1[1] * ans2[2] + ans1[0] * ans2[0];
    p = ans1[0] * ans2[1] + ans1[2] * ans2[2] + ans1[1] * ans2[0];
    cout << y << ' ' << p << ' ' << w;
    return 0;
}