BZOJ 4276: [ONTAK2015]Bajtman i Okrągły Robin [线段树优化建边]

4276: [ONTAK2015]Bajtman i Okrągły Robin

题意:\(n \le 5000\)个区间\(l,r\le 5000\),每个区间可以选一个点得到val[i]的价值,每个点最多选1次,求最大价值


有个显然的\(n^2\)条边的费用流建图(二分图最大权匹配),每个区间一个点连(1,val[i])的边,区间向区间内每个点连边,每个点向t连容量为1的边

提前知道这是线段树优化建图,所以想了一个做法:线段树每个节点拆点限制流量,每个节点都向t连边

其实不用这么麻烦,直接线段树父节点向子节点连INF边,子节点处限制流量就可以了


这种做法利用了线段树"**一个区间分成log段,并且途径最多log个点**"的特点。倍增不能实现这样,因为倍增没有父子关系我没想出如何限制流量

PS:跑得好慢啊40s卡时过,毕竟nlogn个点的费用流...
**咦?这个二分图最大权匹配很特殊啊貌似可以贪心乱搞100ms就过去啊,具体见下一篇**

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define fir first
#define sec second
#define lc x<<1
#define rc x<<1|1
#define mid ((l+r)>>1)
#define lson lc, l, mid
#define rson rc, mid+1, r
typedef long long ll;
const int N=3e4+5, M=4e5+5, INF=1e9;
inline ll read(){
char c=getchar();ll x=0,f=1;
while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
return x*f;
} int n, s, t, tot, l, r;
struct meow{int l, r, val;}a[N];
struct edge{int v, c, f, w, ne;}e[M];
int cnt=1, h[N];
inline void ins(int u, int v, int c, int w) {
e[++cnt]=(edge){v, c, 0, w, h[u]}; h[u]=cnt;
e[++cnt]=(edge){u, 0, 0, -w, h[v]}; h[v]=cnt;
}
void build(int x, int l, int r) { //printf("build %d %d %d\n",x,l,r);
if(l==r) ins(x, t, 1, 0), tot=max(tot, x);
else {
ins(x, lc, INF, 0); build(lson);
ins(x, rc, INF, 0); build(rson);
}
}
void segIns(int x, int l, int r, int ql, int qr, int u) { //printf("segIns %d %d %d\n",x,l,r);
if(ql<=l && r<=qr) ins(u, x, 1, 0);
else {
if(ql<=mid) segIns(lson, ql, qr, u);
if(mid<qr) segIns(rson, ql, qr, u);
}
} int q[N], head, tail, inq[N], d[N];
pair<int, int> pre[N];
inline void lop(int &x) {if(x==N) x=1;}
bool spfa(int s, int t) {
memset(inq, 0, sizeof(inq));
memset(d, 0, sizeof(d));
head=tail=1;
q[tail++]=s; d[s]=0; inq[s]=1;
pre[t].fir = -1;
while(head!=tail) {
int u=q[head++]; inq[u]=0; lop(head);
for(int i=h[u];i;i=e[i].ne) {
int v=e[i].v;
if(e[i].c > e[i].f && d[v]<d[u]+e[i].w) {
d[v]=d[u]+e[i].w;
pre[v] = make_pair(u, i);
if(!inq[v]) q[tail++]=v, inq[v]=1, lop(tail);
}
}
}
return pre[t].fir != -1;
}
int ek(int s, int t) {
int flow=0, cost=0;
while(spfa(s, t)) {
int f=INF, x;
for(int i=t; i!=s; i=pre[i].fir) x=pre[i].sec, f=min(f, e[x].c-e[x].f);
flow+=f; cost+=f*d[t]; //printf("flow %d %d\n",flow,d[t]);
for(int i=t; i!=s; i=pre[i].fir) x=pre[i].sec, e[x].f+=f, e[x^1].f-=f;
}
return cost;
}
int main() {
freopen("in","r",stdin);
n=read();
for(int i=1; i<=n; i++) a[i].l=read(), a[i].r=read()-1, a[i].val=read(), l=min(l, a[i].l), r=max(r, a[i].r);
s=0; t=((r-l+1)<<2)+n+1;
build(1, l, r);
for(int i=1; i<=n; i++) ins(s, tot+i, 1, a[i].val), segIns(1, l, r, a[i].l, a[i].r, tot+i);
int ans=ek(s, t);
printf("%d\n",ans);
}
上一篇:CF Drazil and Date (奇偶剪枝)


下一篇:解决WebService 测试窗体只能用于来自本地计算机的请求