【圖論——2-sat】——wedding婚宴

這道題我們可以很輕易地發現是用2-sat來解答的,2-sat也很容易想出來。於是本蒟蒻就很快地寫了出來。然而接下來的wa什麼的就一直伴隨了我一整個上午,整個人都不好了。
這道題最噁心的是新娘的位置要固定。
!!!!新娘新郎也會鬧矛盾

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
int n,m;
vector <int> g[40001];
bool mark[40001*2];
int s[40001];
int jis=0;
bool dfs(int u)
{
    int hv=u^1;
    if(mark[hv]==1)
    {
        return 0;
    }
    if(mark[u]==1)
    {
        return 1;
    }
    mark[u]=1;
    s[++jis]=u;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(dfs(v)==0)
        {
            return 0;
        }
    }
    return 1;
}
bool solve()
{
    for(int i=0;i<=n-1;i++)
    {
        if(mark[i*2]==0&&mark[(i*2)^1]==0)
        {
            jis=0;
            if(dfs(i*2)==0)//之所以先判斷這個,是因爲要先判斷新娘對面是否爲新郎(mark[i*2]==1時對面爲新郎,所以dfs(i*2)要==1)
            {
                if(i==0)
                {
                    return 0;//如果新娘對面不是新郎那麼退出(return 0)
                }
                while(jis>0)
                {
                    mark[s[jis--]]=0;
                }
                if(dfs((i*2)^1)==0)
                {
                    return 0;
                }
            }
        }
    }
    return 1;
}
int main()
{
    while(1)
    {
        memset(mark,0,sizeof(mark));
        memset(s,0,sizeof(s));
        jis=0;
        for(int i=0;i<=40000;i++)
        {
            g[i].clear();
        }
        scanf("%d %d",&n,&m);
        if(n==0&&m==0)
        {
            return 0;
        }
        for(int i=1;i<=m;i++)
        {
            int x,y;
            char sx,sy;
            scanf("%d%c %d%c",&x,&sx,&y,&sy);
                g[x*2].push_back((y*2)^1);
                g[y*2].push_back((x*2)^1);
            }
            if(sx=='h'&&sy=='w')
            {
                g[x*2].push_back(y*2);
                g[(y*2)^1].push_back((x*2)^1);
            }
            if(sx=='w'&&sy=='h')
            {
                g[(x*2)^1].push_back((y*2)^1);
                g[y*2].push_back(x*2);
            }
            if(sx=='w'&&sy=='w')
            {
                g[(x*2)^1].push_back(y*2);
                g[(y*2)^1].push_back(x*2);
            }
        }
        if(solve()==0)
        {
            printf("bad luck\n");
            continue;
        }
        else
        {
            for(int i=1;i<=n-1;i++)
            {
                if(mark[(i*2)^1]==1)//輸出新娘這一邊的人(1爲丈夫0爲妻子)
                {
                    printf("%dh ",i);
                }
                else
                {
                    printf("%dw ",i);
                }
            }
        }
        printf("\n");
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章