#include <bits/stdc++.h> using namespace std; int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL); int n; cin>>n; vector<int> v; vector<int> v1; vector<pair<int,int>> v2; for(int i=0;i<n;i++) { int x; cin>>x; v.push_back(x); } int y=1; sort(v.begin(),v.end()); for(int i=1;i<n;i++){ if(v[i]==v[i-1]){ y++; } else{ v1.push_back(y); y=1; } } v1.push_back(y); sort(v1.begin(),v1.end()); for(int i=0;i<v1.size();i++){ v2.push_back(make_pair(v1[i],i)); } v2.push_back(make_pair(n*4,v1.size())); for(int i=1;i<=n;i++){ int x=0; for(int j=i;j<=n;j+=i){ x+=(((*upper_bound(v2.begin(),v2.end(),make_pair(j+i-1,n+1))).second-(*lower_bound(v2.begin(),v2.end(),make_pair(j,0))).second)*(j/i))*i; //cout<<((*upper_bound(v2.begin(),v2.end(),make_pair(j+i-1,n+1))).second-(*lower_bound(v2.begin(),v2.end(),make_pair(j,0))).second)*(j/i)*i<<" "<<j<<" "<<i<<endl; } cout<<x<<" "; } return 0; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | #include <bits/stdc++.h> using namespace std; int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL); int n; cin>>n; vector<int> v; vector<int> v1; vector<pair<int,int>> v2; for(int i=0;i<n;i++) { int x; cin>>x; v.push_back(x); } int y=1; sort(v.begin(),v.end()); for(int i=1;i<n;i++){ if(v[i]==v[i-1]){ y++; } else{ v1.push_back(y); y=1; } } v1.push_back(y); sort(v1.begin(),v1.end()); for(int i=0;i<v1.size();i++){ v2.push_back(make_pair(v1[i],i)); } v2.push_back(make_pair(n*4,v1.size())); for(int i=1;i<=n;i++){ int x=0; for(int j=i;j<=n;j+=i){ x+=(((*upper_bound(v2.begin(),v2.end(),make_pair(j+i-1,n+1))).second-(*lower_bound(v2.begin(),v2.end(),make_pair(j,0))).second)*(j/i))*i; //cout<<((*upper_bound(v2.begin(),v2.end(),make_pair(j+i-1,n+1))).second-(*lower_bound(v2.begin(),v2.end(),make_pair(j,0))).second)*(j/i)*i<<" "<<j<<" "<<i<<endl; } cout<<x<<" "; } return 0; } |