acwing::acwing3347类 参考

AcWing 3347. 菊花链 更多...

#include <acwing.h>

静态 Public 成员函数
static int	main (istream &cin, ostream &cout)

详细描述

AcWing 3347. 菊花链

在文件 acwing.h 第 1291 行定义.

成员函数说明

main()

int acwing::acwing3347::main ( istream & cin, ostream & cout )

static

在文件 acwing.cpp 第 4166 行定义.

                                                    {
        int n;
        cin >> n;
        auto *flowers    = new int[n];
        auto *prefix_sum = new int[n];
        auto location    = unordered_map<int, unordered_set<int>>();
        int sum          = 0;
        for(int i = 0; i < n; i++) {
            cin >> flowers[i];
            sum += flowers[i];
            prefix_sum[i] = sum;
            location[flowers[i]].insert(i);
        }
        int ans = 0;
        for(int i = 0; i < n; i++) {
            for(int j = i; j < n; j++) {
                const int len = j - i + 1;
                sum           = prefix_sum[j] - (i - 1 >= 0 ? prefix_sum[i - 1] : 0);
                if(sum % len == 0) {
                    int avg = sum / len;
                    for(const auto flower: location[avg]) {
                        if(i <= flower && flower <= j) {
                            ans++;
                            break;
                        }
                    }
                }
            }
        }
        cout << ans;
        delete[] flowers;
        delete[] prefix_sum;
        return 0;
    }

被这些函数引用 acwing::TEST().

---

该类的文档由以下文件生成:

• acwing.h
• acwing.cpp