4276: [ONTAK2015]Bajtman i Okrągły Robin 线段树优化费用流

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

Time Limit: 40 Sec  Memory Limit: 256 MB
Submit: 490  Solved: 255

[Submit][Status][Discuss]

Description

有n个强盗，其中第i个强盗会在[a[i],a[i]+1]，[a[i]+1,a[i]+2]，...，[b[i]-1,b[i]]这么多段长度为1时间中选出一个时间进行抢劫，并计划抢走c[i]元。作为保安，你在每一段长度为1的时间内最多只能制止一个强盗，那么你最多可以挽回多少损失呢？

Input

第一行包含一个正整数n(1<=n<=5000)，表示强盗的个数。
接下来n行，每行包含三个正整数a[i],b[i],c[i](1<=a[i]<b[i]<=5000,1<=c[i]<=10000)，依次描述每一个强盗。

Output

输出一个整数，即可以挽回的损失的最大值。

Sample Input

4

1 4 40

2 4 10

2 3 30

1 3 20

Sample Output

90

区间加边，想到线段树优化

之前写过线段树优化最短路

那就YY一下线段树优化网络流

听说还可以贪心 QwQ

这个题多路增广优化不是很明显

喜闻乐见1A了

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

typedef long long ll;

inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
return x*f;
}
void print(ll x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=75010,inf=0X3f3f3f3f;

int n,m,root,cnt,S,T=N-1;

int ecnt=1,last
;
struct EDGE{int to,nt,val,ct;}e[N<<3];
inline void readd(int u,int v,int val,int ct)
{e[++ecnt]=(EDGE){v,last[u],val,ct};last[u]=ecnt;}
inline void add(int u,int v,int val,int ct)
{readd(u,v,val,ct);readd(v,u,0,-ct);}

struct seg_tree{int l,r,ls,rs;}tr
;

void build(int &k,int l,int r,int fa)
{
k=++cnt;
tr[k].l=l;tr[k].r=r;
add(fa,k,r-l+1,0);
if(l==r)return ;
int mid=l+r>>1;
build(tr[k].ls,l,mid,k);
build(tr[k].rs,mid+1,r,k);
}

void modify(int k,int x,int y,int pos)
{
int l=tr[k].l,r=tr[k].r;
if(l>=x&&r<=y){add(k,pos,1,0);return ;}
int mid=l+r>>1;
if(y<=mid)modify(tr[k].ls,x,y,pos);
else if(x>mid)modify(tr[k].rs,x,y,pos);
else modify(tr[k].ls,x,y,pos),modify(tr[k].rs,x,y,pos);
}

bool inq
,vis
;
int dis
,q
;
bool spfa()
{
memset(dis,-1,sizeof(dis));
memset(inq,0,sizeof(inq));
memset(vis,0,sizeof(vis));
register int i,j,u,head=0,tail=1;
dis[S]=0;inq[0]=1;q[0]=0;
while(head!=tail)
{
u=q[head++];if(!(head^N))head=0;inq[u]=0;
for(i=last[u];i;i=e[i].nt)
if(e[i].val&&dis[e[i].to]<dis[u]+e[i].ct)
{
dis[e[i].to]=dis[u]+e[i].ct;
if(!inq[e[i].to])
{q[tail++]=e[i].to;inq[e[i].to]=1;if(!(tail^N))tail=0;}
}
}
return dis[T]!=-1;
}

int ans=0;
int dfs(int u,int lim)
{
if(u==T){ans+=dis[u]*lim;return lim;}
vis[u]=1;int res=0;
register int i,tmp;
for(i=last[u];i;i=e[i].nt)
if(e[i].val&&dis[e[i].to]==dis[u]+e[i].ct&&!vis[e[i].to])
{
tmp=dfs(e[i].to,min(e[i].val,lim));
e[i].val-=tmp,e[i^1].val+=tmp;lim-=tmp;res+=tmp;
if(!lim)break;
}
if(!res)dis[u]=-1;
return res;
}

void maxcf()
{while(spfa())dfs(S,inf);}

int main()
{
n=read();
register int i;
static int l
,r
,ct
;
for(i=1;i<=n;i++){l[i]=read();r[i]=read();ct[i]=read();m=max(r[i],m);}
build(root,1,m-1,0);
for(i=1;i<=n;i++)
{
modify(1,l[i],r[i]-1,cnt+i);
add(cnt+i,cnt+i+n,1,ct[i]);add(cnt+i+n,T,1,0);
}
maxcf();print(ans);puts("");
return 0;
}
/*
4
1 4 40
2 4 10
2 3 30
1 3 20

90
*/