洛谷 P4475 巧克力王国 解题报告
P4475 巧克力王国
题目描述
巧克力王国里的巧克力都是由牛奶和可可做成的。但是并不是每一块巧克力都受王国人民的欢迎，因为大家都不喜欢过于甜的巧克力。
对于每一块巧克力，我们设 \(x\) 和 \(y\) 为其牛奶和可可的含量。由于每个人对于甜的程度都有自己的评判标准，所以每个人都有两个参数 \(a\) 和 \(b\) ，分别为他自己为牛奶和可可定义的权重， 因此牛奶和可可含量分别为 \(x\) 和 \(y\) 的巧克力对于他的甜味程度即为 \(ax+by\)。而每个人又有一个甜味限度 \(c\) ，所有甜味程度大于等于 \(c\) 的巧克力他都无法接受。每块巧克力都有一个美味值 \(h\) 。
现在我们想知道对于每个人，他所能接受的巧克力的美味值之和为多少。
输入输出格式
输入格式：
第一行两个正整数 \(n\) 和 \(m\) ，分别表示巧克力个数和询问个数。
接下来\(n\)行，每行三个整数 \(x , y , h\) ，含义如题目所示。
再接下来 \(m\) 行，每行三个整数 \(a , b , c\) ，含义如题目所示。
输出格式：
输出\(m\)行，其中第\(i\)行表示第\(i\)个人所能接受的巧克力的美味值之和。
说明
对于\(100\%\)的数据，\(1\le n,m\le 50000,-10^9\le a_i,b_i,x_i,y_i\le 10^9\)。
kdtree 搞一下就行了，没有插入就不需要重构
然后复杂度是为什么呢（
Code:
#include <cstdio>
#include <algorithm>
#include <cctype>
#define ll long long
const int N=5e4+10;
const int K=2;
const ll inf=1ll<<45;
int read()
{
int x=0,f=1;char c=getchar();
while(!isdigit(c)) f=c=='-'?0:1,c=getchar();
while(isdigit(c)) x=x*10+c-'0',c=getchar();
return f?x:-x;
}
#define ls ch[now][0]
#define rs ch[now][1]
int ch[N][2],L[N][2],R[N][2],p[N][2],val[N],s[N];
ll sum[N],a,b,c;
int n,m,nk,root;
void ckmin(ll &mi,ll yuy){mi=mi<yuy?mi:yuy;}
void ckmax(ll &mx,ll yuy){mx=mx>yuy?mx:yuy;}
using std::min;
using std::max;
void updata(int now)
{
sum[now]=sum[ls]+sum[rs]+val[now];
for(int i=0;i<K;i++)
{
L[now][i]=R[now][i]=p[now][i];
if(ls) L[now][i]=min(L[now][i],L[ls][i]),R[now][i]=max(R[now][i],R[ls][i]);
if(rs) L[now][i]=min(L[now][i],L[rs][i]),R[now][i]=max(R[now][i],R[rs][i]);
}
}
bool cmp(int a,int b){return p[a][nk]<p[b][nk];}
void build(int &now,int l,int r,int k)
{
if(l>r) return;
int mid=l+r>>1;nk=k;
std::nth_element(s+l,s+mid,s+r+1,cmp);
now=s[mid];
build(ls,l,mid-1,k^1),build(rs,mid+1,r,k^1);
updata(now);
}
void get(int now,ll &mi,ll &mx)
{
ll bee=a*L[now][0]+b*L[now][1];
ckmin(mi,bee),ckmax(mx,bee);
bee=a*L[now][0]+b*R[now][1];
ckmin(mi,bee),ckmax(mx,bee);
bee=a*R[now][0]+b*L[now][1];
ckmin(mi,bee),ckmax(mx,bee);
bee=a*R[now][0]+b*R[now][1];
ckmin(mi,bee),ckmax(mx,bee);
}
ll query(int now)
{
if(!now) return 0;
ll mi=inf,mx=-inf;
get(now,mi,mx);
if(mx<c) return sum[now];
if(mi>=c) return 0;
return (a*p[now][0]+b*p[now][1]<c?val[now]:0)+query(ls)+query(rs);
}
int main()
{
n=read(),m=read();
for(int i=1;i<=n;i++)
{
p[i][0]=read(),p[i][1]=read(),sum[i]=val[i]=read();
s[i]=i;
}
build(root,1,n,0);
for(int i=1;i<=m;i++)
{
a=read(),b=read(),c=read();
printf("%lld\n",query(root));
}
return 0;
}
2019.2.5
