Codeforces - 65D - Harry Potter and the Sorting Hat - 简单搜索

https://codeforces.com/problemset/problem/65/D

哈利波特！一种新思路的状压记忆化dfs，记得每次dfs用完要减回去。而且一定是要在dfs外部进行加减！防止在中间return的时候忘记弄回来。用哈希记录状态实现真正的记忆化。

#include<bits/stdc++.h>
using namespace std;
#define ll long long

int n;
char s[];

ll getID(int a[]){
    ll res=;
    for(int i=;i<;i++){
        res=res*+a[i];
    }
    return res;
}

int a[]={};
int c[]={};

set<ll> se;

int dfs(int i){
    ll ID=getID(a);
    if(se.count(ID))
        return ;
    else{
        se.insert(ID);
        //printf("%d-%lld\n",i,ID);
        if(i==n){
            int minnum=*min_element(a,a+);
            for(int j=;j<;j++){
                if(a[j]==minnum){
                    c[j]=;
                }
            }
            return ;
        }
    }

    if(s[i]!='?'){
        switch (s[i]){
        case 'G':
            a[]++;
            dfs(i+);
            a[]--;
            break;
        case 'H':
            a[]++;
            dfs(i+);
            a[]--;
            break;
        case 'R':
            a[]++;
            dfs(i+);
            a[]--;
            break;
        case 'S':
            a[]++;
            dfs(i+);
            a[]--;
        }
    }
    else{
        int minnum=*min_element(a,a+);
        for(int j=;j<;j++){
            if(a[j]==minnum){
                //cout<<"j="<<j<<endl;
                a[j]++;
                dfs(i+);
                a[j]--;
            }
        }
    }
}

int main(){
    scanf("%d",&n);
    scanf("%s",s);
    dfs();

    //printf("%d\n",se.size());

    if(c[])
        puts("Gryffindor");
    if(c[])
        puts("Hufflepuff");
    if(c[])
        puts("Ravenclaw");
    if(c[])
        puts("Slytherin");

}