本文主要是介绍2017 多校3 1003 Kanade's sum,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
http://acm.hdu.edu.cn/showproblem.php?pid=6058
这道题,主要是反过来想,求一段序列的所有非空子序列中第k大的数的和
可以理解为是求每一个数在多少个非空子序列中担任第k大的位置
所以要从当前数的前面和后面分别找出k-1个别当前数大的数
那么就可以从前到后枚举有多少个满足当前数为第k大的序列了
注意 因为是找第k大,所以比当前数小的数可以不用考虑了.
#include<bits/stdc++.h>
using namespace std;
int a[666666];
int rig[666666];
int lef[666666];
int main(){int t;cin>>t;while(t--){int n,k;scanf("%d%d",&n,&k);int i,j;for(i=0;i<n;i++)scanf("%d",&a[i]);long long int ans=0;for(i=0;i<n;i++){int le=1,ri=1;for(j=i+1;j<n;j++){if(ri>k)break;if(a[j]>a[i]){rig[ri++]=j-i;}}if(j>=n){rig[ri]=n-i;}for(j=i-1;j>=0;j--){if(le>k)break;if(a[j]>a[i]){lef[le++]=i-j;}}if(j<=0){lef[le]=i+1;}for(j=0;j<le;j++){if(k-j-1>=ri)continue;int aa=lef[j+1]-lef[j];int bb=rig[k-j]-rig[k-j-1];ans+=(long long int )(aa*bb)*a[i];}}cout<<ans<<endl;}return 0;
}
这篇关于2017 多校3 1003 Kanade's sum的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!