之前做过一个类似的,加个循环就好了
ACcode:
#include<cstdio>
#include<iostream>
#include<algorithm>
#define LL long long
using std::sort;
const int nsize=33333;
struct Line
{
int lx,rx,y,f;
Line(){}
Line(int a,int b,int c,int d): lx(a),rx(b),y(c),f(d){}
bool operator <(const Line &cmp) const{
return y<cmp.y;
}
}line[nsize<<1];
int X[nsize<<1],top,k;
int sum[nsize<<3][11]; //记录覆盖i次及以上的长度
int num[nsize<<3];
int Fin(int key,int len)
{
int l=0,r=len;
while (l<=r)
{
int m=(l+r)>>1;
if (X[m]>key) r=m-1;
else if (X[m]<key) l=m+1;
else return m;
}
return -1;
}
void build(int rt,int l,int r)
{
num[rt]=0;
sum[rt][0]=X[r+1]-X[l];
for (int i=1;i<=k;i++) sum[rt][i]=0;
if (l==r) return ;
int m=(l+r)>>1;
build(rt<<1,l,m);
build(rt<<1|1,m+1,r);
}
void PushUp(int rt,int l,int r)
{
int i,t1=rt<<1,t2=t1|1;
if (num[rt])
{
for (i=1;i<=k;i++)
{
if (num[rt]>=i) sum[rt][i]=X[r+1]-X[l];
else if (l==r) sum[rt][i]=0;
else sum[rt][i]=sum[t1][i-num[rt]]+sum[t2][i-num[rt]];
}
}
else
{
for (i=1;i<=k;i++)
if (l==r) sum[rt][i]=0;
else sum[rt][i]=sum[t1][i]+sum[t2][i];
}
}
void update(int rt,int l,int r,int L,int R,int v)
{
if (L<=l&&r<=R)
{
num[rt]+=v;
PushUp(rt,l,r);
return ;
}
int m=(l+r)>>1;
if (L<=m) update(rt<<1,l,m,L,R,v);
if (R>m) update(rt<<1|1,m+1,r,L,R,v);
PushUp(rt,l,r);
}
int main()
{
LL ans;
int T,cas=0;
int n,m,i,j,L,R;
int x1,x2,y1,y2;
scanf("%d",&T);
while (T--)
{
scanf("%d %d",&n,&k);
for (top=i=0;i<n;i++)
{
scanf("%d %d %d %d",&x1,&y1,&x2,&y2);
x2++,y2++;
line[top]={x1,x2,y1,1}; X[top++]=x1;
line[top]={x1,x2,y2,-1}; X[top++]=x2;
}
sort(X,X+top);
sort(line,line+top);
for (m=0,i=1;i<top;i++)
if (X[i]!=X[i-1]) X[++m]=X[i];
build(1,0,m);
for (ans=i=0;i<top-1;i++)
{
L=Fin(line[i].lx,m);
R=Fin(line[i].rx,m)-1;
if (L<=R) update(1,0,m,L,R,line[i].f);
ans+=1LL*sum[1][k]*(line[i+1].y-line[i].y);
}
printf("Case %d: %lld\n",++cas,ans);
}
return 0;
}