假设有 n 根柱子,现要按下述规则在这 n 根柱子中
依次放入编号为 1,2,3,…的球。
(1)每次只能在某根柱子的 最上面放球。
(2)在同一根柱子中,任何 2 个相邻球的编号之和为完全平方数。
试设计一个算法,计算出在 n 根柱子上最多能放多少个球。例如,在 4 根柱子上最多可放 11 个球。
(1)每次只能在某根柱子的 最上面放球。
(2)在同一根柱子中,任何 2 个相邻球的编号之和为完全平方数。
试设计一个算法,计算出在 n 根柱子上最多能放多少个球。例如,在 4 根柱子上最多可放 11 个球。
对于给定的 n,计算在 n 根柱子上最多能放多少个球。
http://wikioi.com/problem/1234/
最小路径覆盖:找最少的路径,覆盖图中所有的点,每个点只存在于一条路径中。
最小路径覆盖 = 二分图一侧的顶点数 - 最大匹配(最大流)
可以这样理解,当图中没有路径的时候,最小路径=顶点数 - 0,每增加一个匹配,路径就要减少1。
这道题就是将每个数拆成二分图中的两个点i 和 i‘,从源点向i连一条边,从i'向汇点连一条边。如果i+j=完全平方数,就从i向j'连一条边。
每加一个数求一次最小路径,每一次最小路径 = n都是一个可行解,求最大的可行解。
AC代码:每次都重新求最大流(因为oj貌似没有special judge =。=)
//#pragma comment(linker,"/STACK:1024000000,1024000000")
#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef double DB;
typedef long long ll;
typedef pair<int, int> PII;
#define pb push_back
#define MP make_pair
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
const DB eps = 1e-6;
const int inf = ~0U>>1;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1000000007;
const int maxn = 40000 + 10;
///init是初始化要在加边之前初始化,然后调用max_flow(顶点数,边数,源点, 汇点)
const int maxv = 40000 + 10;///顶点个数
const int maxe = 1000000 + 10;///边数
int c[maxe];
struct node{
int v, cap, next;
}edge[maxe];
int head[maxv], cnt;
int n;///n是节点个数,m是边数
int st, ed;///st是源点,ed是汇点
int gap[maxv], h[maxv];
void addedge(int u, int v, ll w){///有向图加边
edge[cnt].v = v; edge[cnt].cap = w; edge[cnt].next = head[u]; head[u] = cnt++;///正向边
edge[cnt].v = u; edge[cnt].cap = 0; edge[cnt].next = head[v]; head[v] = cnt++;///反向边
}
int dfs(int x, int cost){
if(x == ed) return cost; ///当前节点是汇点,直接返回cost
int can = cost, d, minh = n - 1;
for(int i=head[x]; ~i; i=edge[i].next){
int v = edge[i].v, w = edge[i].cap;
if(w > 0){///如果这条边的容量大于0
if(h[v] + 1 == h[x]){///如果这是允许弧
if(can > w) d = w;///如果当前弧的容量小于之前可增广的容量
else d = can;
d = dfs(v, d);///从v开始可增广的容量为d
///更新弧的容量和可增广的容量
edge[i].cap -= d;
edge[i ^ 1].cap += d;
can -= d;
if(h[st] >= n) return cost - can;
if(!can) break;///不能再继续增广
}
if(h[v] < minh) minh = h[v];///更新最小标号
}
}
if(can == cost){///如果没有增广...GAP
gap[h[x]]--;
if(gap[h[x]] == 0) h[st] = n;///存在断层,没有增广路了
h[x] = minh + 1;///重新标记,保证下次再访问的时候有流量
gap[h[x]]++;
}
return cost - can;///在这个点之前可以增广的 - 访问这个点之后可以增广的 = 在这个点增广的容量
}
int max_flow(int N){///SAP+GAP优化
n = N;//m = M;
for(int i=0; i<cnt; i++) c[i] = edge[i].cap;
memset(h, 0, sizeof(h));///h[i]表示i节点的标号
memset(gap, 0, sizeof(gap));///gap[i]表示标号为i的节点个数
gap[0] = n;///初始有n个节点标号为0
int ret = 0;
while(h[st] < n){
ret += dfs(st, inf);
}
return ret;
}
void init(int source, int sink){
memset(head, -1, sizeof(head)); cnt = 0;
st = source; ed = sink;
}
bool vis[maxn];
void Dfs(int x){
vis[x] = true;
for(int i=head[x]; ~i; i=edge[i].next){
if(edge[i].cap > 0 && !vis[edge[i].v]) Dfs(edge[i].v);
}
}
int used[maxn];
void Find(int x, int idx){
used[x] = idx;
int v;
bool flag = false;
for(int i=head[x << 1]; ~i; i=edge[i].next){
v = edge[i].v;
if(edge[i].cap == 0 && v != st){flag = true; break;}
}
if(flag) Find(v >> 1, idx);
}
int N;
int main(){
cin >> N;
int num = 1;
init(0, 1);
int ans = 0, tot;
while(1){
addedge(st, num << 1, 1);
addedge(num << 1 | 1, ed, 1);
for(int i=1; i<num; i++)
if(floor(sqrt((DB)i + num)) == ceil(sqrt((DB)i + num)))
addedge(i << 1, num << 1 | 1, 1);
ans = max_flow((num << 1) + 2);
if(num - ans == N){
memset(used, -1, sizeof(used));
tot = 0;
for(int i=1; i<=num; i++)
if(used[i] == -1) Find(i, tot++);
}
if(num - ans > N) break;
num ++;
for(int i=0; i<cnt; i++) edge[i].cap = c[i];
}
cout << --num << endl;
for(int i=0; i<tot; i++){
bool flag = false;
for(int j=1; j<=n; j++)
if(used[j] == i){
if(!flag){printf("%d", j); flag = true;}
else printf(" %d", j);
}
puts("");
}
return 0;
}
每次在残余网络中求最大流的代码
//#pragma comment(linker,"/STACK:1024000000,1024000000")
#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef double DB;
typedef long long ll;
typedef pair<int, int> PII;
#define pb push_back
#define MP make_pair
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
const DB eps = 1e-6;
const int inf = ~0U>>1;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1000000007;
const int maxn = 40000 + 10;
///init是初始化要在加边之前初始化,然后调用max_flow(顶点数,边数,源点, 汇点)
const int maxv = 40000 + 10;///顶点个数
const int maxe = 1000000 + 10;///边数
int c[maxe];
struct node{
int v, cap, next;
}edge[maxe];
int head[maxv], cnt;
int n;///n是节点个数,m是边数
int st, ed;///st是源点,ed是汇点
int gap[maxv], h[maxv];
void addedge(int u, int v, ll w){///有向图加边
edge[cnt].v = v; edge[cnt].cap = w; edge[cnt].next = head[u]; head[u] = cnt++;///正向边
edge[cnt].v = u; edge[cnt].cap = 0; edge[cnt].next = head[v]; head[v] = cnt++;///反向边
}
int dfs(int x, int cost){
if(x == ed) return cost; ///当前节点是汇点,直接返回cost
int can = cost, d, minh = n - 1;
for(int i=head[x]; ~i; i=edge[i].next){
int v = edge[i].v, w = edge[i].cap;
if(w > 0){///如果这条边的容量大于0
if(h[v] + 1 == h[x]){///如果这是允许弧
if(can > w) d = w;///如果当前弧的容量小于之前可增广的容量
else d = can;
d = dfs(v, d);///从v开始可增广的容量为d
///更新弧的容量和可增广的容量
edge[i].cap -= d;
edge[i ^ 1].cap += d;
can -= d;
if(h[st] >= n) return cost - can;
if(!can) break;///不能再继续增广
}
if(h[v] < minh) minh = h[v];///更新最小标号
}
}
if(can == cost){///如果没有增广...GAP
gap[h[x]]--;
if(gap[h[x]] == 0) h[st] = n;///存在断层,没有增广路了
h[x] = minh + 1;///重新标记,保证下次再访问的时候有流量
gap[h[x]]++;
}
return cost - can;///在这个点之前可以增广的 - 访问这个点之后可以增广的 = 在这个点增广的容量
}
int max_flow(int N){///SAP+GAP优化
n = N;//m = M;
memset(h, 0, sizeof(h));///h[i]表示i节点的标号
memset(gap, 0, sizeof(gap));///gap[i]表示标号为i的节点个数
gap[0] = n;///初始有n个节点标号为0
int ret = 0;
while(h[st] < n){
ret += dfs(st, inf);
}
return ret;
}
void init(int source, int sink){
memset(head, -1, sizeof(head)); cnt = 0;
st = source; ed = sink;
}
bool vis[maxn];
void Dfs(int x){
vis[x] = true;
for(int i=head[x]; ~i; i=edge[i].next){
if(edge[i].cap > 0 && !vis[edge[i].v]) Dfs(edge[i].v);
}
}
int used[maxn];
void Find(int x, int idx){
used[x] = idx;
int v;
bool flag = false;
for(int i=head[x << 1]; ~i; i=edge[i].next){
v = edge[i].v;
if(edge[i].cap == 0 && v != st){flag = true; break;}
}
if(flag) Find(v >> 1, idx);
}
int N;
int main(){
cin >> N;
int num = 1;
init(0, 1);
int ans = 0;
while(1){
addedge(st, num << 1, 1);
addedge(num << 1 | 1, ed, 1);
for(int i=1; i<num; i++)
if(floor(sqrt((DB)i + num)) == ceil(sqrt((DB)i + num)))
addedge(i << 1, num << 1 | 1, 1);
ans += max_flow((num << 1) + 2);
if(num - ans > N) break;
num ++;
}
cout << --num << endl;
memset(used, -1, sizeof(used));
int tot = 0;
for(int i=1; i<=num; i++)
if(used[i] == -1) Find(i, tot++);
for(int i=0; i<tot; i++){
bool flag = false;
for(int j=1; j<=n; j++)
if(used[j] == i){
if(!flag){printf("%d", j); flag = true;}
else printf(" %d", j);
}
puts("");
}
return 0;
}