Optimal Marks

给定一个无向图,每个顶点都有对应值,每条边的权值为val[u] ^ val[u],同时已知其中某些点的值,求其他点的权值使得最终的图的边权和最小?

点权只能是1或者0,边权和最小
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
#include<cctype>
#include<string>
#include<algorithm>
#include<iostream>
#include<ctime> 
#include<map>
#include<set>
using namespace std;
#define MP(x,y) make_pair((x),(y))
#define PB(x) push_back(x)
typedef long long LL;
const int INF=1000111222;
const double INFF=1e100;
const double eps=1e-8;
const int NN=505;
const int MM=3010;
struct G
{
    int v,cap,next;
}E[MM*4+NN*2];
int p[NN],T;
int dd[NN],qw[NN],tp[NN];
int a[NN],b[NN],x[MM],y[MM];
void add(int u,int v,int f)
{
    E[T].v=v;
    E[T].cap=f;
    E[T].next=p[u];
    p[u]=T++;
    E[T].v=u;
    E[T].cap=0;
    E[T].next=p[v];
    p[v]=T++;
}
bool find_path(int st,int en,int n)
{
    int i,u,v,head,tail;
    for(i=0;i<=n;i++)
        dd[i]=-1;
    dd[st]=0;
    qw[head=tail=0]=st;
    while(head<=tail)
    {
        u=qw[head++];
        for(i=p[u];i+1;i=E[i].next)
        {
            v=E[i].v;
            if(dd[v]==-1 && E[i].cap>0)
            {
                dd[v]=dd[u]+1;
                qw[++tail]=v;
            }
        }
    }
    return dd[en]!=-1;
}
int dfs_flow(int u,int &en,int f)
{
    if(u==en || f==0)
        return f;
    int temp,flow=0;
    for( ; tp[u]+1 ; tp[u] = E[tp[u]].next )
    {
        G &e=E[tp[u]];
        if(dd[e.v]==dd[u]+1)
        {
            temp=dfs_flow(e.v,en,min(f,e.cap));
            if(temp>0)
            {
                flow+=temp;
                f-=temp;
                e.cap-=temp;
                E[tp[u]^1].cap+=temp;
                if(f==0)
                    break;
            }
        }
    }
    return flow;
}
void dinic(int st,int en,int n)
{
    int i;
    while( find_path(st,en,n) )
    {
        for(i=0;i<=n;i++)
            tp[i]=p[i];
        dfs_flow(st,en,INF);
    }
}
int main()
{
    int cas,n,m;
    int i,j,k,u,v;
    scanf("%d",&cas);
    while(cas--)
    {
        scanf("%d%d",&n,&m);
        for(i=0;i<m;i++)
            scanf("%d%d",&x[i],&y[i]);
        memset(a,-1,sizeof(a));
        memset(b,0,sizeof(b));
        scanf("%d",&k);
        while(k--)
        {
            scanf("%d%d",&i,&j);
            a[i]=j;
            b[i]=j;
        }
        for(i=0;i<31;i++)
        {
            memset(p,-1,sizeof(p));
            T=0;
            for(j=0;j<m;j++)
            {
                u=x[j];
                v=y[j];
                if(a[u]>=0 && a[v]>=0 && (a[u]%2)==(a[v]%2))
                    continue;
                add(u,v,1);
                add(v,u,1);
            }
            for(j=1;j<=n;j++)
            {
                if(a[j]>=0)
                {
                    if(a[j]&1)
                        add(0,j,INF);
                    else
                        add(j,n+1,INF);
                    a[j]>>=1;
                }
            }
            dinic(0,n+1,n+1);
            for(j=1;j<=n;j++)
            {
                if(dd[j]>=0 && a[j]==-1)
                {
                    b[j]=b[j]|(1<<i);
                }
            }
        }
        for(i=1;i<=n;i++)
        {
            printf("%d\n",b[i]);
        }
    }
    return 0;
}
发布了44 篇原创文章 · 获赞 35 · 访问量 792

猜你喜欢

转载自blog.csdn.net/huangziguang/article/details/104482333
今日推荐