鳶一折纸
2024-11-18 16:57:55
比较一眼的费用流。
跑一遍费用流即可。
#include <bits/stdc++.h>
using namespace std;
const int N = 2e3 + 10;
namespace MCMF
{
const int MCN = 3010, MCE = 1e6 + 10;
struct node
{
int u, v, w, c, nex;
} e[MCE];
int cnt = 1, cost;
int dis[MCN], vis[MCN], ly[MCN], cur[MCN];
void addedge(int u, int v, int w, int c)
{
e[++cnt] = {u, v, w, c, ly[u]};
ly[u] = cnt;
}
void add(int u, int v, int w, int c) { addedge(u, v, w, c), addedge(v, u, 0, -c); }
bool spfa(int s, int t)
{
memset(dis, 127, sizeof(dis));
memcpy(cur, ly, sizeof(ly));
int minn = dis[s];
vis[s] = 1, dis[s] = 0;
queue<int> q;
q.push(s);
while (q.size())
{
int now = q.front();
q.pop();
vis[now] = 0;
for (int i = ly[now]; i; i = e[i].nex)
{
int to = e[i].v;
if (e[i].w && dis[to] > dis[now] + e[i].c)
{
dis[to] = dis[now] + e[i].c;
if (!vis[to])
q.push(to), vis[to] = 1;
}
}
}
return dis[t] != minn;
}
int dinic(int now, int t, int flow)
{
if (now == t)
return flow;
vis[now] = 1;
int maxn = 0;
for (int &i = cur[now]; i && maxn < flow; i = e[i].nex)
{
int v = e[i].v;
if (!vis[v] && e[i].w && dis[v] == dis[now] + e[i].c)
{
int res = dinic(v, t, min(e[i].w, flow - maxn));
if (res)
cost += res * e[i].c, e[i].w -= res, e[i ^ 1].w += res, maxn += res;
}
}
vis[now] = 0;
return maxn;
}
int mcmf(int s, int t)
{
int res = 0;
while (spfa(s, t))
{
int x = dinic(s, t, 1e9);
while (x)
res += x, x = dinic(s, t, 1e9);
}
return res;
}
}
using namespace MCMF;
namespace prime
{
bool visp[N];
int p[N >> 2];
void eulur(int n)
{
for (int i = 2; i <= n; ++i)
{
if (!visp[i])
p[++p[0]] = i;
for (int j = 1; j <= p[0] && i * p[j] <= n; ++j)
{
visp[i * p[j]] = 1;
if (i % p[j] == 0)
break;
}
}
}
}
using namespace prime;
int n, S, T;
signed main()
{
ios::sync_with_stdio(0);
cin >> n, eulur(2000), S = p[0] + n + 2, T = S + 1;
for (int i = 1; i <= p[0]; ++i)
add(i + n + 1, T, 1, 0);
add(n + 1, T, n, 0);
for (int i = 1, a; i <= n; ++i)
{
cin >> a;
for (int j = 1; j <= p[0]; ++j)
{
if (a % p[j])
continue;
add(i, n + j + 1, 1, a / p[j]);
}
add(S, i, 1, 0), add(i, n + 1, 1, a);
}
(void)mcmf(S, T);
cout << cost << endl;
return 0;
}