ural1156. Two Rounds(二分染色+dfs)

好久没有写过题了,有点生疏啊,这题坑了好久......写了个裸的爆搜TLE了,然后将他分类后,只对有2个点以上的连通分量爆搜就过了,只含一个点的连通分量可以随便在两个round里选,所以可以不搜,最后随便放就行。
#include 
#include 
#include 
#include 
#include 
using namespace std;
#define N 105
#define M 210
struct node
{
    int to,next;
} e[M];
int head[N],esum;
int n,m;
inline void init()
{
    memset(head,-1,sizeof(head));
    esum=0;
}
inline void addedge(int u,int v)
{
    e[esum].to=v,e[esum].next=head[u],head[u]=esum++;
}
int belong[N],cnt;
bool err,vis[N];
vectorlst[N][2];
vector::iterator it;
void dfs(int id,int flag)
{
    if(err) return ;
    if(!belong[id])
    {
        belong[id]=flag;
        lst[cnt/2][flag%2].push_back(id);
    }
    else if(belong[id]!=flag)
    {
        err=1;
        return ;
    }
    int col=1-(flag-cnt)+cnt;
    for(int i=head[id]; ~i; i=e[i].next)
    {
        int to=e[i].to;
        if(!belong[to])
            dfs(to,col);
        else if(belong[to]!=col)
        {
            err=1;
            return ;
        }
    }
}
bool havFind;
bool ans[N];
int sum;
void findAns(int id,int s1,int s2)
{
    if(havFind) return ;
    if(s1>n||s2>n) return ;
    if(id==cnt/2)
    {
        havFind=1;
        sum=s1;
        return ;
    }
    if(lst[id][0].size()+lst[id][1].size()>1)
    {
        ans[id]=0;
        findAns(id+1,s1+lst[id][0].size(),s2+lst[id][1].size());
        if(havFind) return ;
        ans[id]=1;
        findAns(id+1,s1+lst[id][1].size(),s2+lst[id][0].size());
        if(havFind) return ;
    }
    else findAns(id+1,s1,s2);
}
int main()
{
    int a,b;
    while(~scanf("%d%d",&n,&m))
    {
        init();
        for(int i=0; i

你可能感兴趣的:(图论,dfs/bfs)