2017 Multi-University Training Contest - Team 3 Kanade's sum

最新推荐文章于 2020-05-03 14:33:09 发布

UFO___

最新推荐文章于 2020-05-03 14:33:09 发布

阅读量366

点赞数

分类专栏：算法文章标签：链表 C++ 算法

本文链接：https://blog.csdn.net/UFO___/article/details/76595273

版权

算法专栏收录该内容

48 篇文章 0 订阅

订阅专栏

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

但是边界处理比较恶心（我比较水）。

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

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

然后暴力枚举最小的那个元素，唯一的一种可能是里面所有元素都比他大，这样只需枚举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