hdoj 4183 Pahom on Water 【基础最大流】

逃离我推掉我的手 2022-08-11 08:58 150阅读 0赞

题目：[hdoj 4183 Pahom on Water][]

题意：题目有点长，读懂了就是个水的最大流，每次从789开始到400，走的话必须是两个圆相交而且频率递增的，每个点只走一次，求有没有满足这样条件的。

分析：题目读懂就比较水了。直接按照题目意思建图，初始点和结束点容量为2，其他点只走一次容量为1，然后求最大流。

AC代码：

#include <cstdio>
    #include <cstring>
    #include <iostream>
    #include <string>
    #include <algorithm>
    #include <vector>
    #include <queue>
    #include <cmath>
    using namespace std;
    #define Del(a,b) memset(a,b,sizeof(a))
    const int N = 1020;
    const int inf = 0x3f3f3f3f;
    const double esp = 1e-9;
    int n,m;
    struct Node
    {
        int from,to,cap,flow;
    };
    struct C
    {
        double pp;
        int x,y,z;
    };
    C a[N];
    vector<int> v[N];
    vector<Node> e;
    int vis[N];  //构建层次图
    int cur[N];
    void add_Node(int from,int to,int cap)
    {
        e.push_back((Node){from,to,cap,0});
        e.push_back((Node){to,from,0,0});
        int tmp=e.size();
        v[from].push_back(tmp-2);
        v[to].push_back(tmp-1);
    }
    bool bfs(int s,int t)
    {
        Del(vis,-1);
        queue<int> q;
        q.push(s);
        vis[s] = 0;
        while(!q.empty())
        {
            int x=q.front();
            q.pop();
            for(int i=0;i<v[x].size();i++)
            {
                Node tmp = e[v[x][i]];
                if(vis[tmp.to]<0 && tmp.cap>tmp.flow)  //第二个条件保证
                {
                    vis[tmp.to]=vis[x]+1;
                    q.push(tmp.to);
                }
            }
        }
        if(vis[t]>0)
            return true;
        return false;
    }
    int dfs(int o,int f,int t)
    {
        if(o==t || f==0)  //优化
            return f;
        int a = 0,ans=0;
        for(int &i=cur[o];i<v[o].size();i++) //注意前面 ’&‘,很重要的优化
        {
            Node &tmp = e[v[o][i]];
            if(vis[tmp.to]==(vis[o]+1) && (a = dfs(tmp.to,min(f,tmp.cap-tmp.flow),t))>0)
            {
                tmp.flow+=a;
                e[v[o][i]^1].flow-=a; //存图方式
                ans+=a;
                f-=a;
                if(f==0)  //注意优化
                    break;
            }
        }
        return ans;  //优化
    }
    
    int dinci(int s,int t)
    {
        int ans=0;
        while(bfs(s,t))
        {
            Del(cur,0);
            int tm=dfs(s,inf,t);
            ans+=tm;
        }
        return ans;
    }
    bool solve(int x,int y)
    {
        int l=(a[x].x-a[y].x)*(a[x].x-a[y].x)+(a[x].y-a[y].y)*(a[x].y-a[y].y);
        int r=(a[x].z+a[y].z)*(a[x].z+a[y].z);
        if(l<=r&&a[x].pp>a[y].pp) //相交
            return true;
        return false;
    }
    int main()
    {
        //freopen("Input.txt","r",stdin);
        int T;
        scanf("%d",&T);
        for(int cas=1;cas<=T;cas++)
        {
            scanf("%d",&n);
            for(int i=1;i<=n;i++)
                scanf("%lf%d%d%d",&a[i].pp,&a[i].x,&a[i].y,&a[i].z);
            int s,t;
            for(int i=1;i<=n;i++)
            {
                if(abs(a[i].pp-789.0)<=esp)
                {
                    s = i;
                    add_Node(s,i+n,2);
                }
                else if(abs(a[i].pp-400.0<=esp))
                {
                    t = i+n;
                    add_Node(i,t,2);
                }
                else
                    add_Node(i,i+n,1);
                for(int j=1;j<=n;j++)
                {
                    if(i!=j && solve(i,j))
                        add_Node(i+n,j,1);
                }
            }
            int ans=dinci(s,t);
            printf(ans==2?"Game is VALID\n":"Game is NOT VALID\n");
            for(int i=0;i<=n*2;i++)
                v[i].clear();
            e.clear();
        }
        return 0;
    }

[hdoj 4183 Pahom on Water]: http://acm.hdu.edu.cn/showproblem.php?pid=4183