https://loj.ac/problem/6003
变化了下的最小路径覆盖,要注意边数巨多。。。
#include <iostream>
#include <stdio.h>
#include <queue>
#include <cstring>
using namespace std;
int biao[100];
const int maxn = 10009;
int head[maxn];
int cur[maxn];
int tot = 0;
struct edge{
int v,nex,w;
}e[maxn*20];
void addedge(int u,int v,int w){
e[tot] = (edge){v,head[u],w};
head[u] = tot++;
e[tot] = (edge){u,head[v],w};
head[v] = tot++;
}
int deep[maxn];
int nowball;
bool bfs(int S,int T){
for(int i=0;i<=2*nowball+1;i++){
deep[i] = 0;
}
queue<int> q;
deep[S] = 1;
q.push(S);
while(!q.empty()){
int now = q.front();
q.pop();
for(int i=head[now];i!=-1;i=e[i].nex){
int v = e[i].v;
int w = e[i].w;
if(deep[v]!=0 || w<=0) continue;
deep[v] = deep[now]+1;
if(v==T) return deep[T];
q.push(v);
}
}
return deep[T];
}
int dfs(int now,int T,int maxflow){
if(now==T) return maxflow;
int all =0;
for(int &i=cur[now];i!=-1 &&all <maxflow;i=e[i].nex){
int v = e[i].v;
int w = e[i].w;
if(deep[v]!=deep[now]+1 ||w<=0) continue;
int tt = dfs(v,T,min(maxflow-all,w));
e[i].w-=tt;
e[i^1].w+=tt;
all+=tt;
}
return all;
}
int dinic(int S,int T){
int ret = 0;
for(int i=0;i<=2*nowball+1;i++){
cur[i] = head[i];
}
while(bfs(S,T)){
ret+=dfs(S,T,0x3f3f3f3f);
for(int i=0;i<=2*nowball;i++){
cur[i] = head[i];
}
}
return ret;
}
int vis[maxn*2];
void print(int now){
if(now<=1 || now%2==1) return;
vis[now] = 1;
printf("%d ",now/2);
for(int i=head[now];i!=-1;i=e[i].nex){
int v = e[i].v;
int w = e[i].w;
if(w<=0){
print(v^1);
}
}
}
int main()
{
for(int i=0;i<100;i++){
biao[i] = i*i;
}
memset(head,-1,sizeof(head));
int n;
scanf("%d",&n);
int kkk = 0;
nowball = 0;
while(true){
nowball++;
for(int i=0;i<100;i++){
int from = biao[i]-nowball;
if(from<1 || from>=nowball) continue;
addedge(from*2,nowball*2+1,1);
}
addedge(0,nowball*2,1);
addedge(nowball*2+1,1,1);
kkk+=dinic(0,1);
if(nowball-kkk<=n) continue;
else{
break;
}
}
nowball--;
printf("%d\n",nowball);
for(int i=1;i<=nowball;i++){
if(vis[2*i]==0) {
print(2*i);
printf("\n");
}
}
return 0;
}