题目描述:有n个人,给出所属学校和实力值,k个人才能组成一队,每个学校都按照实力值和尽可能大的方法派出队伍,问k从1~n,分别求所有学校所有队伍的实力值和。
把人按照学校分开,实力值从小到大排序做前缀和,对某个学校有m个人,派出的实力值和sum[m]-sum[m%k]。累加进ans[k]即可。
复杂度O(Σ nlogn)
#include<bits/stdc++.h>
#define ll long long
using namespace std;
struct orz
{
ll u;
ll v;
}a[200010];
ll n;
ll sum[200010],ans[200010];
bool my(orz a,orz b)
{
if(a.u==b.u) return a.v<b.v;
else return a.u<b.u;
}
void ops(ll l,ll r)
{
for(ll i=l;i<=r;++i)
{
sum[i-l+1]=sum[i-l]+a[i].v;
}
ll m=r-l+1;
for(ll i=1;i<=m;++i)
{
ans[i]+=sum[m]-sum[m%i];
}
}
int main()
{
ll t;
cin>>t;
while(t--)
{
cin>>n;
for(ll i=1;i<=n;++i)
{
cin>>a[i].u;
}
for(ll i=1;i<=n;++i)
{
cin>>a[i].v;
}
sort(a+1,a+1+n,my);
ll k=1;
for(ll i=1;i<=n;++i)
{
while(a[i].u==a[k].u&&k<=n) ++k;
ops(i,k-1);
i=k-1;
}
for(ll i=1;i<=n;++i)
{
cout<<ans[i]<<' ';
ans[i]=0;
}
cout<<endl;
}
//getchar();getchar();
return 0;
} 


京公网安备 11010502036488号