Codeforces Round #568 (Div. 2)

G1. Playlist for Polycarp (easy version)

theme:n首歌，给定每首歌的时长t1与分类gi,问共有几种不同的选歌方式，使得每首歌最多被选1次，总的时长为T且相邻两首歌的类别不同。1≤n≤15,1≤T≤225，1<=gi<=3

solution:像这种n很小的选择问题考虑状态压缩dp。用dp[s][i]表示状态为s，最后一首歌的类别为i时的选择方法，则每一个dp[s][i]可以用来更新再选一首歌的情况，即：dp[s^(1<<j)][i]+=dp[s][i]。把g转化到0~2,dp[0][3]=1用来初始化dp[0][0],dp[0][1],dp[0][2]后面的dp[s][3]不会用到

#include<bits/stdc++.h>
using namespace std;
#define far(i,t,n) for(int i=t;i<n;++i)
typedef long long ll;
typedef unsigned long long ull;
using namespace std;

const int maxs=1<<15;
int p=1e9+7;
int dp[maxs+5][4];
int t[20],g[20];

int main()
{
    int n,T;
    cin>>n>>T;
    far(i,0,n)
        cin>>t[i]>>g[i],g[i]--;

    int ans=0;
    dp[0][3]=1;
    for(int s=0;s<=maxs-1;++s)
        for(int i=0;i<=3;++i)
        {
            for(int j=0;j<n;++j)
            {
                if(((s&(1<<j))==0)&&(g[j]!=i))
                {
                    dp[s^(1<<j)][g[j]]=(dp[s^(1<<j)][g[j]]+dp[s][i])%p;
                }
            }
            int x=0;
            far(j,0,n)
                if(s&(1<<j))
                    x+=t[j];
            if(x==T)
                ans=(ans+dp[s][i])%p;
        }
    cout<<ans<<endl;
}