Codeforces - Strange Set

题目链接:Codeforces - Strange Set


不难看出是一个最大权闭合子图。
但是我们考虑优化连边,显然每个点对于同一种权值来说,只会连前面最近的。


AC代码:

#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
//#define int long long
using namespace std;
const int inf=0x3f3f3f3f;
const int N=3e3+10,M=N*100;
int n,s,t,h[N],a[N],b[N],res;
int head[N],nex[M],w[M],to[M],tot=1;
inline void ade(int a,int b,int c){to[++tot]=b; nex[tot]=head[a]; w[tot]=c; head[a]=tot;}
inline void add(int a,int b,int c){ade(a,b,c); ade(b,a,0);}
int bfs(){
	queue<int> q; q.push(s); memset(h,0,sizeof h); h[s]=1;
	while(q.size()){
		int u=q.front(); q.pop();
		for(int i=head[u];i;i=nex[i]) if(!h[to[i]]&&w[i]) h[to[i]]=h[u]+1,q.push(to[i]);
	}
	return h[t];
}
int dfs(int x,int f){
	if(x==t) return f; int fl=0;
	for(int i=head[x];i&&f;i=nex[i]){
		if(w[i]&&h[to[i]]==h[x]+1){
			int mi=dfs(to[i],min(w[i],f));
			w[i]-=mi,w[i^1]+=mi,fl+=mi,f-=mi;
		}
	}
	if(!fl) h[x]=-1;
	return fl;
}
int dinic(int res=0){
	while(bfs()) res+=dfs(s,inf);
	return res;
}
signed main(){
	cin>>n; t=n+1;
	for(int i=1;i<=n;i++) cin>>a[i];
	for(int i=1;i<=n;i++) cin>>b[i];
	for(int i=2;i<=n;i++){
		int vis[110]={0};
		for(int j=i-1;j>=1;j--) if(a[i]%a[j]==0&&!vis[a[j]]){
			vis[a[j]]=1; add(i,j,inf);
		}
	}
	for(int i=1;i<=n;i++)
		if(b[i]>0) res+=b[i],add(s,i,b[i]);
		else add(i,t,-b[i]);
	cout<<res-dinic();
	return 0;
}
已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 酷酷鲨 设计师:CSDN官方博客 返回首页