CF1185G1 Playlist for Polycarp (easy version)

状压dp康复训练  AFO太久了连状压都忘了

一开始我想的状态是选到了第i个曲子,之前已经选了a1, a2, a3个类型分别为1, 2, 3的曲子的个数 ,所用时间为t

但后来通过第二个样例发现这么做与遍历曲子的顺序有关,也就是说打乱输入给的曲子的顺序答案会完全不一样  然后我就不会了 

之后看了题解才恍然大悟,尼玛这不就是旅行商问题(TSP)吗

设i为当前选了那些曲子的状态,j是放在最后的曲子的类型,直接转移就好了

初始化的时候令f[0][0]=1,一个曲子都不选的时候j就等于0,这样i=0的状态转移其他状态时可以接任意类型的曲子

最后统计答案时只有i状态选的曲子的总时长等于T时,才计入答案

#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define db double
#define rep(x,a,b) for(int x=(a);x<=(b);x++)
#define per(x,a,b) for(int x=(a);x>=(b);x--)
#define scf(a) scanf("%d",&a)
#define scfll(a) scanf("%lld",&a)
#define scfdb(a) scanf("%lf",&a)
#define ptf(a) printf("%d",a)
#define ptfll(a) printf("%lld",a)
#define ptfdb(x,a) printf("%x.lf",a)
#define ptfsp(a) printf("%d ",a)
#define ptfllsp(a) printf("%lld ",a)
#define ptfdbsp(x,a) printf("%x.lf ",a)
#define pli(a,b) make_pair(a,b)
#define pb push_back
#define el puts("")
#define pi 3.1415926
//ios::sync_with_stdio(false);
using namespace std;
const ll mod=1e9+7;
const int maxn=2e5+5;
ll f[1<<16][4];
int t[16],g[16];
int main(){
    int n,T;
    scf(n);scf(T);
    rep(i,1,n) scf(t[i]),scf(g[i]);
    f[0][0]=1;
    rep(i,0,(1<<n)-1){
        rep(k,0,3)
            rep(j,1,n)
                if(((i>>(j-1))&1)==0&&g[j]!=k)
                    (f[i|(1<<(j-1))][g[j]]+=f[i][k])%=mod;
    }
    ll ans=0;
    rep(i,0,(1<<n)-1){
        int tt=0;
        ll res=0;
        rep(j,1,n)
            if((i>>(j-1))&1) tt+=t[j];
        if(tt==T){
            rep(k,1,3) (ans+=f[i][k])%=mod;
        }
    }
    cout<<ans;
}

猜你喜欢

转载自blog.csdn.net/ghu99999/article/details/115197549
今日推荐