UVA 11983 Weird Advertisement
继续种树~~
之前ZXL童鞋说那俩题不好做，我就先隔过去啦~然后就UVA这个题啦。
和之前差不多，求至少K次覆盖的区域面积（虽然是求点数，但是可以转化为面积哈，因为我用的不是点树哈）。
和3642差不多，这里的K最大是10，所以可以采用一样的方法。关键是怎么找规律，总不能写K个if。。else吧。。
其实想一下就能推出来的，如果当前cover 为 x，那么这个区间覆盖 x + y 次的长度为当前区间子区间覆盖y次的长度和。结点里存的是slen[i]是覆盖次数为i的长度，如果i == K存的是大于等于K的次数。
超过K次的都按K次计算。
注意初始化，0次的话，是总长度，因为这个WA了N次。。。
其他叶节点什么的，注意下细节就好。
P.S. ZXL童鞋的代码比我的短了好多！！！学习之~他的结点存的是slen[i] 覆盖大于等于 i 的次数。这样的话，所有更新一个循环就可以搞定了。
我的：
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <limits.h>
#include <string.h>
#include <string>
#include <algorithm>
#define MID(x,y) ( ( x + y ) >> 1 )
#define L(x) ( x << 1 )
#define R(x) ( x << 1 | 1 )
#define FOR(i,s,t) for(int i=(s); i<(t); i++)
#define BUG puts("here!!!")
#define STOP system("pause")
#define file_r(x) freopen(x, "r", stdin)
#define file_w(x) freopen(x, "w", stdout)
using namespace std;
const int MAX = 60010;
typedef long long LL;
struct Sline{ int x,y1,y2,flag;};
struct Tnode{ // 一维线段树
int l, r, cover, slen[12];
int len() { return r - l;}
int mid() { return MID(l,r);}
bool in(int ll,int rr) { return l >= ll && r <= rr; }
void lr(int ll,int rr){ l = ll; r = rr;}
};
Tnode node[MAX<<2];
Sline l[MAX];
int y[MAX], cnty, cnt, K;
void add_line(int x1,int y1,int x2,int y2,int &cnt)
{
l[cnt].x = x1; l[cnt].y1 = y1; l[cnt].y2 = y2;
l[cnt].flag = 1;
y[cnt++] = y1;
l[cnt].x = x2; l[cnt].y1 = y1; l[cnt].y2 = y2;
l[cnt].flag = -1;
y[cnt++] = y2;
}
void Build(int t,int l,int r)
{
node[t].lr(l,r);
node[t].cover = 0;
memset(node[t].slen+1, 0, K*sizeof(int));
node[t].slen[0] = y[node[t].r] - y[node[t].l];
if( node[t].len() == 1 )
return ;
int mid = MID(l,r);
Build(L(t),l,mid);
Build(R(t),mid,r);
}
void Updata_len(int t)
{
int cc = node[t].cover;
memset(node[t].slen, 0, (K+1)*sizeof(int));
if( cc >= K )
{
node[t].slen[K] = y[node[t].r] - y[node[t].l];
return ;
}
if( node[t].len() == 1 )
{
node[t].slen[cc] = y[node[t].r] - y[node[t].l];
return ;
}
FOR(i, 0, K+1)
{
int d = i + cc;
if( d >= K ) d = K;
if( d == K )
node[t].slen[d] += (node[L(t)].slen[i] + node[R(t)].slen[i]);
else
node[t].slen[d] = node[L(t)].slen[i] + node[R(t)].slen[i];
}
int sum = 0;
FOR(i, 1, K+1)
sum += node[t].slen[i];
node[t].slen[0] = y[node[t].r] - y[node[t].l] - sum;
}
void Updata(int t, Sline p)
{
if( p.y1 <= y[node[t].l] && p.y2 >= y[node[t].r] )
{
node[t].cover += p.flag;
Updata_len(t);
return ;
}
if( node[t].len() == 1 ) return ;
int mid = node[t].mid();
if( p.y1 < y[mid] ) Updata(L(t), p);
if( p.y2 > y[mid] ) Updata(R(t), p);
Updata_len(t);
}
bool cmp(Sline a, Sline b)
{
if( a.x == b.x )
return a.flag > b.flag;
return a.x < b.x;
}
LL solve(int n)
{
LL ans = 0;
sort(y, y+n);
cnty = unique(y, y+n) - y;
sort(l, l+n, cmp);
Build(1, 0, cnty-1);
Updata(1, l[0]);
FOR(i, 1, n)
{
ans += node[1].slen[K] * 1ll * (l[i].x - l[i-1].x);
Updata(1, l[i]);
}
return ans;
}
int main()
{
int ind = 1, ncases, n, x1, y1, x2, y2;
scanf("%d", &ncases);
while( ncases-- )
{
cnt = 0;
scanf("%d%d", &n, &K);
FOR(i, 0, n)
{
scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
x2++; y2++;
add_line(x1, y1, x2, y2, cnt);
}
if( K > n )
{
printf("Case %d: 0\n", ind++);
continue;
}
LL ans = solve(cnt);
printf("Case %d: %lld\n", ind++, ans);
}
return 0;
}
改进后的：Build里面也不用给slen[0]赋初值了
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <limits.h>
#include <string.h>
#include <string>
#include <algorithm>
#define MID(x,y) ( ( x + y ) >> 1 )
#define L(x) ( x << 1 )
#define R(x) ( x << 1 | 1 )
#define FOR(i,s,t) for(int i=(s); i<(t); i++)
#define BUG puts("here!!!")
#define STOP system("pause")
#define file_r(x) freopen(x, "r", stdin)
#define file_w(x) freopen(x, "w", stdout)
using namespace std;
const int MAX = 60010;
typedef long long LL;
struct Sline{ int x,y1,y2,flag;};
struct Tnode{ // 一维线段树
int l, r, cover, slen[12];
int len() { return r - l;}
int mid() { return MID(l,r);}
bool in(int ll,int rr) { return l >= ll && r <= rr; }
void lr(int ll,int rr){ l = ll; r = rr;}
};
Tnode node[MAX<<2];
Sline l[MAX];
int y[MAX], cnty, cnt, K;
void add_line(int x1,int y1,int x2,int y2,int &cnt)
{
l[cnt].x = x1; l[cnt].y1 = y1; l[cnt].y2 = y2;
l[cnt].flag = 1;
y[cnt++] = y1;
l[cnt].x = x2; l[cnt].y1 = y1; l[cnt].y2 = y2;
l[cnt].flag = -1;
y[cnt++] = y2;
}
void Build(int t,int l,int r)
{
node[t].lr(l,r);
node[t].cover = 0;
memset(node[t].slen+1, 0, K*sizeof(int));
// node[t].slen[0] = y[node[t].r] - y[node[t].l];
if( node[t].len() == 1 )
return ;
int mid = MID(l,r);
Build(L(t),l,mid);
Build(R(t),mid,r);
}
void Updata_len(int t)
{
int cc = node[t].cover;
FOR(i, 0, K+1)
{
if( cc >= i )
node[t].slen[i] = y[node[t].r] - y[node[t].l];
else
if( node[t].len() == 1 )
node[t].slen[i] = 0;
else
node[t].slen[i] = node[L(t)].slen[i-cc] + node[R(t)].slen[i-cc];
}
}
void Updata(int t, Sline p)
{
if( p.y1 <= y[node[t].l] && p.y2 >= y[node[t].r] )
{
node[t].cover += p.flag;
Updata_len(t);
return ;
}
if( node[t].len() == 1 ) return ;
int mid = node[t].mid();
if( p.y1 < y[mid] ) Updata(L(t), p);
if( p.y2 > y[mid] ) Updata(R(t), p);
Updata_len(t);
}
bool cmp(Sline a, Sline b)
{
if( a.x == b.x )
return a.flag > b.flag;
return a.x < b.x;
}
LL solve(int n)
{
LL ans = 0;
sort(y, y+n);
cnty = unique(y, y+n) - y;
sort(l, l+n, cmp);
Build(1, 0, cnty-1);
Updata(1, l[0]);
FOR(i, 1, n)
{
ans += node[1].slen[K] * 1ll * (l[i].x - l[i-1].x);
Updata(1, l[i]);
}
return ans;
}
int main()
{
int ind = 1, ncases, n, x1, y1, x2, y2;
scanf("%d", &ncases);
while( ncases-- )
{
cnt = 0;
scanf("%d%d", &n, &K);
FOR(i, 0, n)
{
scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
x2++; y2++;
add_line(x1, y1, x2, y2, cnt);
}
if( K > n )
{
printf("Case %d: 0\n", ind++);
continue;
}
LL ans = solve(cnt);
printf("Case %d: %lld\n", ind++, ans);
}
return 0;
}