2017 Multi-University Training Contest

来源:互联网 发布:医院网络咨询 编辑:程序博客网 时间:2024/06/10 12:19

这个题目要反过来计算每个数字对那些区间的贡献。

但是边界处理比较恶心(我比较水)。

至于过程官方说的比较明白了。

先把所有元素存入一个链表中,但是用一个数组把每个元素的大小顺序记录下来(从小到大)。

然后暴力枚举最小的那个元素,唯一的一种可能是里面所有元素都比他大,这样只需枚举k次。

然后把这个元素删除,在枚举链表里面当前最小的元素,那么所有元素又都比他大。

然后用记录的下标计算该数对多少个区间产生了贡献,将结果乘这个元素大小 在累加即可。

就是这里从大到小排序再枚举比较难想。

#include<iostream>        #include<cstdio>        using namespace std;                typedef struct node        {        int v;        int pre;        int nex;        int pos;}node;node ar[500005];long long ord[500005],znum[500005],ynum[500005];                int main(){    //freopen("fuck.txt","r",stdin);    int T;    scanf("%d",&T);    for(int Case=1;Case<=T;Case++)    {    int n,k;    scanf("%d%d\n",&n,&k);    long long ans=0;    for(int i=0;i<=n+1;i++)    {    if(1<=i&&i<=n) {scanf("%d",&ar[i].v);ord[ar[i].v]=i;}    ar[i].nex=i+1;    ar[i].pre=i-1;ar[i].pos=i;}for(int i=1;i<=n;i++){int z=ar[ord[i]].pre,pz=ord[i],y=ar[ord[i]].nex,py=ord[i];for(int j=0;j<=k;j++) znum[j]=ynum[j]=0;for(int j=0;j<k&&z>=0;j++){znum[j]=-(ar[z].pos-ar[pz].pos);pz=z;z=ar[z].pre;}for(int j=0;j<k&&y<=n+1;j++){ynum[j]=ar[y].pos-ar[py].pos;py=y;y=ar[y].nex;}for(int u=0,v=k-1;u<k&&v>=0;u++,v--){ans+=(znum[u]*ynum[v]*i);}node t=ar[ord[i]];ar[t.pre].nex=t.nex;ar[t.nex].pre=t.pre;}printf("%lld\n",ans);}return 0


原创粉丝点击