洛谷2765魔术球问题

题目链接:魔术球问题

首先明确了“每个球只能被用一次”这个条件,我们就要考虑拆点了

接下来我们一个一个枚举球,对于当前枚举到的球考虑它和之前的球是否有连接关系,若有则将它们连起来

然后跑最大流,当我们在残量网络中跑出的最大流不为0时,就说明新球可以与之前的一根柱子建立对应关系(他可以通过之前的球“串”)起来,否则我们就要单独为它开一个新柱子,柱子数用一个变量储存,当它\(>n\)时直接退出

然后就是简(e)单(xin)的输出了

#include<iostream>
#include<string>
#include<string.h>
#include<stdio.h>
#include<algorithm>
#include<math.h>
#include<vector>
#include<queue>
#include<map>
using namespace std;
#define rep(i,a,b) for (i=a;i<=b;i++)
typedef long long ll;
#define maxd 1e9+7
struct node{
    int from,to,nxt,flow;
}sq[200100];
int all=1,dep[100100],head[100100],cur[100100],n=1e5+2,m,s,t;
bool vis[100100];
void add(int u,int v,int w)
{
    all++;sq[all].from=u;sq[all].to=v;sq[all].nxt=head[u];sq[all].flow=w;head[u]=all;
    all++;sq[all].from=v;sq[all].to=u;sq[all].nxt=head[v];sq[all].flow=0;head[v]=all;
}

bool bfs()
{
    queue<int> q;int i;
    memset(vis,0,sizeof(vis));
    vis[s]=1;q.push(s);dep[s]=0;
    while (!q.empty())
    {
        int u=q.front();q.pop();
        for (i=head[u];i;i=sq[i].nxt)
        {
            int v=sq[i].to;
            if ((!vis[v]) && (sq[i].flow))
            {
                vis[v]=1;dep[v]=dep[u]+1;q.push(v);
            }
        }
    }
    if (!vis[t]) return 0;
    for (i=1;i<=n;i++) cur[i]=head[i];
    return 1;
}

    int dfs(int now,int to,int lim)
    {
        if ((!lim) || (now==to)) return lim;
        int i,sum=0;
        for (i=cur[now];i;i=sq[i].nxt)
        {
            cur[now]=i;int v=sq[i].to;
            if (dep[now]+1==dep[v])
            {
                int f=dfs(v,to,min(lim,sq[i].flow));
                if (f)
                {
                    lim-=f;sum+=f;
                    sq[i].flow-=f;
                    sq[i^1].flow+=f;
                    if (!lim) break;
                }
            }
        }
        return sum;
    }

    int work_flow()
    {
        int sum=0;
        while (bfs()) sum+=dfs(s,t,maxd);
        return sum;
    }

int ans[60],cnt=0;

int read()
{
    int x=0,f=1;char ch=getchar();
    while ((ch<'0') || (ch>'9')) {if (ch=='-') f=-1;ch=getchar();}
    while ((ch>='0') && (ch<='9')) {x=x*10+(ch-'0');ch=getchar();}
    return x*f;
}

void out(int u)
{
    vis[u]=1;
    int i;
    for (i=head[u];i;i=sq[i].nxt)
    {
        int v=sq[i].to;
        if ((v==s) || (v==t) || (sq[i].flow) || (vis[v])) continue;
        printf("%d ",v/2);
        int tmp=v/2;tmp*=2;
        out(tmp);
    }
}

void init()
{
    m=read();s=1e5+1;t=1e5+2;
}

void work()
{
    int now=0;
    while (cnt<=m)
    {
        now++;int i;
        add(s,now*2,1);
        add(now*2+1,t,1);
        for (i=sqrt(now)+1;i*i<now*2;i++)
        {
            int tmp=i*i-now;
            add(tmp*2,now*2+1,1);
        }
        int tmp=work_flow();
        if (!tmp) ans[++cnt]=now;
        if (cnt>m) break;
    }
    now--;int i;
    printf("%d\n",now);
    memset(vis,0,sizeof(vis));
    for (i=1;i<=m;i++)
    {
        printf("%d ",ans[i]);
        out(ans[i]*2);
        printf("\n");
    }
}

int main()
{
    init();
    work();
    return 0;
}
posted @ 2018-12-24 10:38  EncodeTalker  阅读(148)  评论(0编辑  收藏  举报