最小路径覆盖问题
,这道题,一开始我花了100多ms,看了下排行榜,,发现个10多ms的。看了很久他的代码,才明白怎么回事。同时更加加深了我对匈牙利算法的理解。
我从他的代码中学到的是
(1) 若多加几条边,并不需要重新跑一边匈牙利,只需要从没匹配的点开始dfs就可以了,
(2)若因为多加了一个点,从而多加了边,只需要从这个点dfs一次就好了,,不需要重新跑匈牙利
果然,要多看别人代码啊~
不过这题还有一个神一般的贪心算法,大概是
第一个球球当然是放在第一个柱子里面了
以后每一个球都从第一个柱子开始判断是否能放进去,如果可以的话就放啦,如果不行就新开一个啦。。不能再开就完啦。
然而,,我没有证明出来这个为什么正确,,有谁知道的,,求留个言啊
#include<iostream>
#include<map>
#include<stdio.h>
#include<algorithm>
#include<queue>
#include<set>
#include<string.h>
#include<string>
#include<vector>
#define MX 3333
#define M 1555
#define INF 0x3f3f3f3f
#define mem(x,y) memset(x,y,sizeof(x))
#define FIN freopen("input.txt","r",stdin)
using namespace std;
int nx,ny,n;
int head[MX],rear;
struct Edge
{
int nxt,to;
} edge[22*MX];
void edge_init()
{
mem(head,-1);
rear=0;
}
void edge_add(int a,int b)
{
edge[rear].to=b;
edge[rear].nxt=head[a];
head[a]=rear++;
}
bool vis[MX];
int matchxy[MX],matchyx[MX];
bool hungarian_dfs(int u)
{
for(int i=head[u]; ~i; i=edge[i].nxt)
{
int v=edge[i].to;
if(vis[v]) continue;
vis[v]=1;
if(matchyx[v]==-1||hungarian_dfs(matchyx[v]))
{
matchyx[v]=u;
matchxy[u]=v;
return 1;
}
}
return 0;
}
bool vvis[3333]= {0};
void init()
{
for(int i=1; i*i<=2*1555; i++) vvis[i*i]=1;
}
int main()
{
init();
while(cin>>n)
{
edge_init();
nx=0;
mem(matchxy,-1);
mem(matchyx,-1);
int ret=0;
while(nx-ret<=n)
{
nx++;
for(int i=1; i<nx; i++) if(vvis[i+nx]) edge_add(nx,M+i);
mem(vis,0);
ret+=hungarian_dfs(nx);
}
nx--;
printf("%d\n",nx);
for(int i=1; i<=nx; i++) if(matchxy[i]==-1)
{
vector<int> ans;
ans.push_back(i);
int u=i,v;
while((v=matchyx[u+M])!=-1)
{
ans.push_back(v);
u=v;
}
for(int i=0; i<ans.size(); i++) printf("%d%c",ans[i],i==ans.size()-1?'\n':' ');
}
}
return 0;
}