题目链接:[SNOI2019]通信


很明显暴力建图,边数达到1e6的级别,费用流必然TLE。

然而连线都是到某个区间连线,所以我们可以想到线段树优化建图。

但是:

我们看一下这个图,可以发现增广的路径长度十分短?所以zkw费用流跑得很快。于是加上剪枝暴力一发!!!


AC代码:

#pragma GCC optimize(2)
#include<bits/stdc++.h>
//#define int long long
using namespace std;
typedef long long ll;
const int N=2e3+10,M=3e6+10,inf=0x3f3f3f3f;
int n,W,st[N],vis[N],s,t,a[N];
int head[N],nex[M],to[M],flow[M],tot=1; ll res,w[M],d[M]; queue<int> q;
inline void ade(int a,int b,int c,int d){
	to[++tot]=b; nex[tot]=head[a]; w[tot]=d; flow[tot]=c; head[a]=tot;
}
inline void add(int a,int b,int c,int d){ade(a,b,c,d); ade(b,a,0,-d);}
inline bool spfa(){
	memset(st,0,sizeof st); memset(d,0x3f,sizeof d); q.push(s); d[s]=0; vis[s]=0;
	while(q.size()){
		int u=q.front();	q.pop();	vis[u]=0;
		for(int i=head[u];i;i=nex[i]){
			if(flow[i]&&d[to[i]]>d[u]+w[i]){
				d[to[i]]=d[u]+w[i];	
				if(!vis[to[i]])	q.push(to[i]),vis[to[i]]=1;
			}
		} 
	}
	return d[t]<0x3f3f3f3f3f3f3f3f;
}
int dfs(int x,int f){
	if(x==t)	return res+=d[t]*f,f;
	st[x]=1;	int fl=0;
	for(int i=head[x];i&&f;i=nex[i]){
		if(!st[to[i]]&&flow[i]&&d[to[i]]==d[x]+w[i]){
			int mi=dfs(to[i],min(flow[i],f));
			flow[i]-=mi,flow[i^1]+=mi,fl+=mi,f-=mi;
		}
	}
	return fl;
}
signed main(){
	cin>>n>>W; t=n*2+1;
	for(int i=1;i<=n;i++)	cin>>a[i],add(s,i,1,0),add(i,t,1,W),add(i+n,t,1,0);
	for(int i=1;i<=n;i++)	for(int j=1;j<i;j++){
		if(abs(a[i]-a[j])<W)	add(i,j+n,1,abs(a[i]-a[j]));
	}
	while(spfa())	dfs(s,inf);
	cout<<res<<endl;
	return 0;
}