LA-3713-TwoSAT

題目大意:有ABC三個任務需要分配給n個宇航員,任務C沒有限制,年齡大於等於平均年齡的可以分配A任務,否則是B任務,有m對宇航員互相討厭,他們不能顧分配相同的方案,問是否可以找出一種方案。

題目解析:如果相互討厭的話,那麼肯定不能夠同時分配C,並且不能分配A和B,如果一個A一個B則不需要考慮;

AC代碼:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<vector>
#include<stack>
using namespace std;
const int maxn=200010;
const int maxm = 100010;
int first[maxn];  
int vv[maxm*4],nxt[maxm*4],S[maxm*4];  
int e,c,n,m;  
bool vis[maxn];  
void addedge(int u,int v)  
{  
    vv[e] = v;  nxt[e] = first[u];  first[u] = e++;  
}  
  
bool dfs(int u) 
{  
    if(vis[u^1])    return 0;  
    if(vis[u])  return 1;  
    vis[u] = 1;  
    S[c++] = u;  
    for(int i = first[u];i != -1;i = nxt[i])  
    {  
        int v = vv[i];  
        if(!dfs(v)) return false;  
    }  
    return true;  
}  
  
bool Judge()  
{  
    for(int i = 0;i < n*2;i+=2)  
    {  
        if(!vis[i] && !vis[i+1])  
        {  
            c = 0;  
            if(!dfs(i))  
            {  
                while(c > 0) vis[S[--c]] = 0;  
                if(!dfs(i+1))   return false;  
            }  
        }  
    }  
    return true;  
}  
int old[maxn],eld[maxn];  
int main()
{
	while(scanf("%d%d",&n,&m)!=EOF&&(n+m))
	{
		memset(first,-1,sizeof(first));
		e=0;
		memset(vis,0,sizeof(vis));
		int sum=0;
		for(int i=0;i<n;i++)
		{
			scanf("%d",&old[i]);
			sum+=old[i];
		}
		for(int i=0;i<n;i++)
			eld[i]=old[i]*n>=sum?1:0;
		for(int i=0;i<m;i++)
		{
			int u,v;
			scanf("%d%d",&u,&v);
			u--;
			v--;
			addedge(2*u+1,2*v);
			addedge(2*v+1,2*u);
			if(eld[u]==eld[v])
				addedge(2*u,2*v+1),addedge(2*v,2*u+1);
		} 
		if(Judge())
		{
			for(int i=0;i<n;i++)
			{
				if(vis[2*i+1]==1)
					printf("C\n");
				else if(eld[i])
					printf("A\n");
				else 
					printf("B\n");
			}
		}
		else 
			printf("No solution.\n");
	}
	return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章