Submission #73061281

---

Source Code Expand

#include <iostream>
#include <vector>
#include <unordered_map>
#include <algorithm>
#include <map>
using namespace std;
vector<long long> get_divisors(long long s) {
    vector<long long> divisors;
    if (s == 0) return divisors;
    map<long long, int> factors;
    for (long long i = 2; i * i <= s; ++i) {
        while (s % i == 0) {
            factors[i]++;
            s /= i;
        }
    }
    if (s > 1) {
        factors[s]++;
    }
    divisors.push_back(1);
    for (auto& p : factors) {
        long long prime = p.first;
        int exp = p.second;
        int sz = divisors.size();
        long long current = 1;
        for (int e = 1; e <= exp; ++e) {
            current *= prime;
            for (int j = 0; j < sz; ++j) {
                divisors.push_back(divisors[j] * current);
            }
        }
    }
    sort(divisors.begin(), divisors.end());
    auto last = unique(divisors.begin(), divisors.end());
    divisors.erase(last, divisors.end());
    return divisors;
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int N;
    cin >> N;
    vector<long long> A(N);
    long long sum_S = 0;
    long long max_A = 0;
    unordered_map<long long, int> cnt_map;
    for (int i = 0; i < N; ++i) {
        cin >> A[i];
        sum_S += A[i];
        if (A[i] > max_A) {
            max_A = A[i];
        }
        cnt_map[A[i]]++;
    }
    vector<long long> divisors = get_divisors(sum_S);
    vector<long long> ans;
    for (long long L : divisors) {
        if (L < max_A) {
            continue;
        }
        long long K = sum_S / L;
        if (K < (N + 1) / 2 || K > N) {
            continue;
        }
        int required = 2 * K - N;
        int actual = cnt_map.count(L) ? cnt_map[L] : 0;
        if (actual == required) {
            ans.push_back(L);
        }
    }
    sort(ans.begin(), ans.end());
    for (int i = 0; i < ans.size(); ++i) {
        if (i > 0) {
            cout << " ";
        }
        cout << ans[i];
    }
    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function 'int main()':
./Main.cpp:72:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   72 |     for (int i = 0; i < ans.size(); ++i) {
      |                     ~~^~~~~~~~~~~~

Judge Result