CF1473F. Strange Set
题意:
给你一个长度为
n
n
n的数组
a
[
1...
n
]
a[1...n]
a[1...n],选每个元素的代价为
b
[
1...
n
]
b[1...n]
b[1...n]。如果你选择
a
[
i
]
a[i]
a[i],那么对于每个
a
[
j
]
,
j
∈
[
1
,
i
−
1
]
a[j],j\in[1,i-1]
a[j],j∈[1,i−1]如果是
a
[
i
]
a[i]
a[i]的因子,那么也必须选。
问选出的集合最大权值和为多少。
思路:
选择某一个元素,一定条件的必须选上,显然是最大权闭合子图。
但是由于空间限制,我们要优化建边。对于
a
[
i
]
a[i]
a[i]来说,前面相同的因子只要连第一个就行了,因为第一个肯定会连向第二个。
其余就是最大流模板了。
代码:
#include<bits/stdc++.h>
#define pii pair<int,int>
#define ll long long
#define cl(x,y) memset(x,y,sizeof(x))
#define loop(x,y,z) for(x=y;x<=z;x++)
#define reve(x,y,z) for(x=y;x>=z;x--)
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define all(x) x.begin(),x.end()
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
#define INF 1e18
const int N=1e6+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const double eps=1e-8;
const double pi=acos(-1);
using namespace std;
int a[3010];
struct edge
{
int u,v,w;
}e[N];
int head[3010],len=1,dep[3010],s,t;
void add(int u,int v,int w)
{
e[++len]={head[u],v,w};
head[u]=len;
}
void inc(int u,int v,int w)
{
add(u,v,w);
add(v,u,0);
}
int dfs(int u,int f,int t)
{
int ans=0,i;
if(u==t)
return f;
for(i=head[u];i && f;i=e[i].u)
{
int v=e[i].v,w=e[i].w;
if(dep[v]==dep[u]+1 && w)
{
int sum=dfs(v,min(f,w),t);
e[i].w-=sum;
e[i^1].w+=sum;
f-=sum;
ans+=sum;
}
}
if(!ans)
dep[u]=-2;
return ans;
}
int bfs(int s,int t)
{
queue<int> q;
cl(dep,0);
dep[s]=1;
q.push(s);
while(!q.empty())
{
int u=q.front(),i;
q.pop();
for(i=head[u];i;i=e[i].u)
{
int v=e[i].v,w=e[i].w;
if(w && !dep[v])
{
dep[v]=dep[u]+1;
q.push(v);
}
}
}
return dep[t];
}
int dinic(int s,int t)
{
int ans=0;
while(bfs(s,t))
ans+=dfs(s,inf,t);
return ans;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int n,i,j;
cin>>n;
s=0;
t=n+1;
for(i=1;i<=n;i++)
cin>>a[i];
for(i=1;i<=n;i++)
{
int vis[110]={0};
for(j=i-1;j>=1;j--)
if(a[i]%a[j]==0 && !vis[a[j]])
{
vis[a[j]]=1;
inc(i,j,inf);
}
}
int sum=0;
for(i=1;i<=n;i++)
{
int w;
cin>>w;
if(w>0)
{
inc(s,i,w);
sum+=w;
}
else
inc(i,t,-w);
}
cout<<sum-dinic(s,t)<<endl;
return 0;
}