2020冬季PAT甲级---Chemical Equation

7-4 Chemical Equation

这道题纯模拟吧…
关键在于存储方程式的状态，这里采用状态压缩，然后排序的时候注意一下就好了…
其他没什么，题目看懂都能做…

AC代码

#include <iostream>
#include <algorithm>
#include <map>
#include <vector>
using namespace std;
int go_int(string s,int le,int ri)
{
    int ans = 0;
    ans = (s[le]-'0')*10 + s[ri]-'0';
    return ans;
}
const int MAX = 105;

map<int,int> mp;
vector<int> all_way[MAX];
int pro[MAX];
int rea[MAX];
int n;

int vis;

vector<int> ans,temp;
bool cmp(int a,int b)
{
	for(int i=0;i<21;i++)
    {
        if((a & (1<<i)) && ((b&(1<<i)) == 0))
        return false;
        else if((a & (1<<i)) == 0 && (b&(1<<i)))
        return true;
    }
}
bool ok(int now)
{
	for(int i=0;i<n;i++)
	{
		if(((1<<i) & now) && ((vis & (1<<i)) == 0))
		return false;
	}
	
	for(int i=0;i<n;i++)
	{//删除1 
		if((1<<i)&now)
		vis -= (1<<i);
	}
	return true;
} 
void dfs(int cur,int m)
{
	if(cur == m)
	{
		ans = temp;
		return ;
	}
	
	int res = pro[cur];
	for(int i=0;i<all_way[res].size();i++)
	{
		int k = all_way[res][i];
		if(ok(k))
		{
			temp.push_back(k);
			dfs(cur+1,m);
			temp.pop_back();
			for(int i=(1<<20);i;i>>=1)
			{
				if(i&k)
				vis += i;
			}
		}
	}
}
void print(int res,int k)
{
	int f = 0;
	for(int i=0;i<n;i++)
	if((1<<i)&k)
	{
		if(!f)	printf("%02d",rea[i]),f = 1;
		else	printf(" + %02d",rea[i]);
	}
	printf(" -> %02d\n",res);
}
int main()
{
	scanf("%d",&n);
	for(int i=0;i<n;i++)
	{
		int x;
		scanf("%d",&x);
		//mp[x] = i;
		rea[i] = x;
	}
	sort(rea,rea+n);
	for(int i=0;i<n;i++)
	{
		mp[rea[i]] = i;
		all_way[rea[i]].push_back(1<<i);
	}
	
	vis = (1<<n)-1;
	
	int m;
	scanf("%d",&m);
	for(int i=0;i<m;i++)
	scanf("%d",&pro[i]);
	
	int k;
	scanf("%d",&k);
	getchar();
	while(k--)
	{
		string s;
        getline(cin,s);
        int len = s.length();
        int res = go_int(s,len-2,len-1);
        
        int l = 0;//当前状态 
        for(int i=0;i<=len-6;i++)
        {
        	if(s[i] >= '0' && s[i] <= '9')
        	{
        		int x = go_int(s,i,i+1);
        		i++;
        		l += (1<<mp[x]);
			}
		}
		all_way[res].push_back(l); 
	}
	for(int i=0;i<m;i++)
	{
		int res = pro[i];
		sort(all_way[res].begin(),all_way[res].end(),cmp);
	}
	dfs(0,m);
	for(int i=0;i<m;i++)
	print(pro[i],ans[i]);
		
	return 0;
}