Codeforces Round #601 (Div. 2) E1 Send Boxes to Alice (Easy Version)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5+;
int a[N];
int n;
bool prime(int x) {//判断是否为质数
    for(int i = ; i*i <= x; i++) {
        if(x%i == ) return false;
    }
    return true;
}
ll solve(int x) {
    vector<int>b;
    ll ans = ;
    for(int i = ; i <= n; i++) {
        if(a[i] ==  && b.size() < x) b.push_back(i);//存放位置
        if(b.size() == x) {
            for(int j = ; j < b.size(); j++) {
                ans += (ll)abs(b[j]-b[x/]);
            }
            b.clear();
        }
    }
    return ans;
}
int main() {
    int sum = ;
    scanf("%d",&n);
    for(int i = ; i <= n; i++) {
        scanf("%d",a+i);
        sum += a[i];
    }
    if(sum == ) {
        printf("-1\n");
        return ;
    }
    ll ans = 0x3f3f3f3f3f;
    for(int i = ; i <= sum; i++) {
        if(sum%i==&&prime(i)) {//此处用质因子优化，不加优化也行。
            ans = min(ans,solve(i));
        }
    }
    printf("%lld\n",ans);
    return ;
}