【JZOJ A组】Melancholy

Description

DX3906星系，Melancholy星上，我在勘测这里的地质情况。
我把这些天来已探测到的区域分为N组，并用二元组(D,V)对每一组进行标记：其中D为区域的相对距离，V为内部地质元素的相对丰富程度。
在我的日程安排表上有Q项指派的计划。每项计划的形式是类似的，都是“对相对距离D在[L,R]之间的区域进行进一步的勘测，并在其中有次序地挑出K块区域的样本进行研究。”采集这K块的样品后，接下来在实验中，它们的研究价值即为这K块区域地质相对丰富程度V的乘积。
我对这Q项计划都进行了评估：一项计划的评估值P为所有可能选取情况的研究价值之和。
但是由于仪器的原因，在一次勘测中，这其中V最小的区域永远不会被选取。
现在我只想知道这Q项计划的评估值对2^32取模后的值，特殊地，如果没有K块区域可供选择，评估值为0。

Input

第一行给出两个整数，区域数N与计划数Q。
第二行给出N个整数，代表每一块区域的相对距离D。
第三行给出N个整数，代表每一块区域的内部地质元素的相对丰富程度V。
接下来的Q行，每一行3个整数，代表相对距离的限制L,R，以及选取的块数K。

Output

输出包括Q行，每一行一个整数，代表这项计划的评估值对2^32取模后的值。

Sample Input

5 3
5 4 7 2 6
1 4 5 3 2
6 7 1
2 6 2
1 8 3

Sample Output

5
52
924

Data Constraint

思路

发现k不大，而且是个区间问题，不妨考虑线段树。

线段树维护区间最小值和区间k=1…6的答案

合并时f[k]=f[i]*f[k-j]

代码

#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
using namespace std;
typedef unsigned int u32;

const int maxn=1000005;
const u32 inf=4294967295u;

int n,q;
int k;
struct node
{
		u32 d,v;
}a[maxn],L,R;
bool cmp(const node &a,const node &b) {return a.d < b.d;}

struct power
{
	u32 val[7];
	friend power operator +(power a,power b)
	{
		power c;
		for(int i=1; i<=6; i++) c.val[i]=a.val[i]+b.val[i];
		for(int i=1; i<=6; i++)
			for(int j=1; j < i; j++)
				c.val[i] += a.val[j]*b.val[i-j];
		return c;
	}
}Node[maxn],A[3],Ans;

struct Min
{
	u32 val,pos;
	friend Min operator +(Min a,Min b)
	{
		Min c=(Min){inf,0};
		if(a.val < c.val) c=a;
		if(b.val < c.val) c=b;
		return c;
	}
}Val[maxn],res_min;


void build(int i,int l,int r)
{
	Val[i]=(Min){inf,0};
	for(int j=1; j<=6; j++) Node[i].val[j]=0;
	if(l == r)
	{
		Node[i].val[1]=a[l].v;
		Val[i]=(Min){a[l].v,l};
		return;
	}
	int mid=l+r >> 1;
	build(i << 1,l,mid); build(i << 1 | 1,mid+1,r);
	Node[i]=Node[i << 1]+Node[i << 1 | 1];
	Val[i]=Val[i << 1]+Val[i << 1 | 1];
}

void find(int i,int l,int r,int L,int R)
{
	if(L>R) return;
	if(L<=l&&r<=R)
	{
		res_min=res_min+Val[i];
		return;
	}
	int mid=l+r >> 1;
	if(L<=mid) find(i << 1,l,mid,L,R);
	if(mid+1<=R) find(i << 1 | 1,mid+1,r,L,R);
}

void query(int i,int l,int r,int L,int R,int opt)
{
	if(L>R) return;
	if(L<=l&&r<=R)
	{
		A[opt]=A[opt]+Node[i];
		return;
	}
	int mid=l+r>>1;
	if(L<=mid) query(i<<1,l,mid,L,R,opt);
	if(mid+1<=R) query(i<<1|1,mid+1,r,L,R,opt);
}


void solve(int k)
{
	int LL=lower_bound(a+1,a+n+1,L,cmp)-a;
	int RR=upper_bound(a+1,a+n+1,R,cmp)-a-1;
	if(LL>=RR) {printf("0\n"); return;}

	res_min=(Min){inf,0};
	find(1,1,n,LL,RR);

	for(int i=1; i<=6; i++) A[1].val[i]=A[2].val[i]=0;
	query(1,1,n,LL,res_min.pos-1,1);
	query(1,1,n,res_min.pos+1,RR,2);

	Ans=A[1]+A[2];
	for(u32 i=1; i<=k; i++) Ans.val[k]*=i;
	printf("%u\n",Ans.val[k]);
}

int main()
{
	scanf("%d%d",&n,&q);
	for(int i=1; i<=n; i++) scanf("%u",&a[i].d);
	for(int i=1; i<=n; i++) scanf("%u",&a[i].v);
	sort(a+1,a+n+1,cmp);

	build(1,1,n);
	
	while(q--)
	{
		scanf("%u%u%u",&L.d,&R.d,&k);
		solve(k);
	}
}