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

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

题意:(n le 5000)个区间(l,rle 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
",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
",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
",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
",ans);
}

原文地址:https://www.cnblogs.com/candy99/p/6641962.html