动态规划2

动态规划2及练习题

前导知识：

背包dp+状压dp相关知识参考OI Wiki之动态规划板块

传送门

1. [NOIP2005 普及组] 采药（记忆化搜索）

传送门

#include <iostream>
#include <cstring>
using namespace std;
int n,t;
int tcost[103],mget[103];
int mem[103][1003];
int INF=0x7ffffff;
int dfs(int pos,int tleft){
    if(mem[pos][tleft]!=-1) return mem[pos][tleft];
    if(pos==n+1) return mem[pos][tleft]=0;
    int dfs1,dfs2=-INF;
    dfs1=dfs(pos+1,tleft);
    if(tleft>=tcost[pos]) dfs2=dfs(pos+1,tleft-tcost[pos])+mget[pos];
    return mem[pos][tleft]=max(dfs1,dfs2);
}
int main(){
    memset(mem,-1,sizeof(mem));
    cin>>t>>n;
    for(int i=1;i<=n;i++)
        cin>>tcost[i]>>mget[i];
    cout<<dfs(1,t)<<endl;
    return 0;
}

2. 魔法照片（模拟+排序）

传送门

#include <cstdio>
#include <algorithm>
using namespace std;
int e[10],n,k;
struct node{
    int w;
    int num;
    int d;
}p[20010];
int cmp(const node &a,const node &b){
    if(a.w!=b.w) return a.w>b.w;
    return a.num<b.num;
}
int main(){
    scanf("%d %d",&n,&k);
    for(int i=0;i<10;i++)
        scanf("%d",&e[i]);
    for(int i=0;i<n;i++){
        scanf("%d",&p[i].w);
        p[i].num=i+1;
    }
    sort(p,p+n,cmp);
    for(int i=0;i<n;i++){
        p[i].d=i%10;
        p[i].w+=e[p[i].d];
    }
    sort(p,p+n,cmp);
    for(int i=0;i<k;i++)
        printf("%d ",p[i].num);
}

3. 通天之分组背包

传送门

#include <iostream>
#include <algorithm>
using namespace std;
int m,n,t1;
int w[10001],c[10001];
int t[205][205];
int cnt[10001];
int dp[10001];
int x; /* 组数编号 */
int main(){
    cin>>m>>n;
    for(int i=1;i<=n;i++){
        cin>>w[i]>>c[i]>>x;
        t1=max(t1,x);
        cnt[x]++;
        t[x][cnt[x]]=i;
    }
    for(int k=1;k<=t1;k++){
        for(int i=m;i>=0;i--){
            for(int j=1;j<=cnt[k];j++)
                if(i>=w[t[k][j]])
                    dp[i]=max(dp[i],dp[i-w[t[k][j]]]+c[t[k][j]]);
        }
    }
    cout<<dp[m];
    return 0;
}

4. 樱花

传送门

#include <cstdio>
#include <algorithm>
using namespace std;
int sx,sy,ex,ey,n,dp[1010];
int a[10005],b[10005],t[10005];
int index;
struct node{
    int w,v;
}list[1000005];
inline void pre(){
    for(int i=1;i<=n;i++){
        int c=1;
        while(t[i]-c>=0){
            list[++index].w=c*a[i];
            list[index].v=c*b[i];
            t[i]-=c;
            c*=2;
        }
        list[++index].w=a[i]*t[i];
        list[index].v=b[i]*t[i];
    }
}
int main(){
    scanf("%d:%d %d:%d %d",&sx,&sy,&ex,&ey,&n);
    int t1=(ex*60+ey)-(sx*60+sy);
    for(int i=1;i<=n;i++){
        scanf("%d %d %d",&a[i],&b[i],&t[i]);
        if(!t[i]) t[i]=999999;
    }
    pre();
    for(int i=1;i<=index;i++){
        for(int j=t1;j>=list[i].w;j--)
            dp[j]=max(dp[j],dp[j-list[i].w]+list[i].v);
    }
    printf("%d",dp[t1]);
    return 0;
}

5. 宝物筛选（二进制优化）

传送门

#include <cstdio>
#include <algorithm>
using namespace std;
int W,n,dp[10005];
int a[10005],b[10005],t[10005];
int index;
struct node{
    int w,v;
}list[1000005];
inline void pre(){
    for(int i=1;i<=n;i++){
        int c=1;
        while(t[i]-c>0){
            list[++index].w=c*b[i];
            list[index].v=c*a[i];
            t[i]-=c;
            c*=2;
        }
        list[++index].w=b[i]*t[i];
        list[index].v=a[i]*t[i];
    }
}
int main(){
    scanf("%d %d",&n,&W);
    for(int i=1;i<=n;i++){
        scanf("%d %d %d",&a[i],&b[i],&t[i]);
    }
    pre();
    for(int i=1;i<=index;i++){
        for(int j=W;j>=list[i].w;j--)
            dp[j]=max(dp[j],dp[j-list[i].w]+list[i].v);
    }
    printf("%d",dp[W]);
    return 0;
}

6. 吃奶酪

传送门

#include <cstdio>
#include <cstring>
#include <cmath>
#define min(a,b) (((a)<(b))?(a):(b))
double a[20][20];
double x[20],y[20];
double dp[18][34000];
int n;
double dis(int v,int w){
    return sqrt((x[v]-x[w])*(x[v]-x[w])+(y[v]-y[w])*(y[v]-y[w]));
}
int main(){
    int i,j,k;
    double ans;
    memset(dp,127,sizeof dp);
    ans=dp[0][0];
    scanf("%d",&n);
    for(i=1;i<=n;i++){
        scanf("%lf %lf",&x[i],&y[i]);
    }
    x[0]=0;y[0]=0;
    for(i=0;i<=n;i++){
        for(j=i+1;j<=n;j++){
            a[i][j]=dis(i,j);
            a[j][i]=a[i][j];
        }
    }
    for(i=1;i<=n;i++){
        dp[i][1<<(i-1)]=a[0][i];
    }
    for(k=1;k<(1<<n);k++){
        for(i=1;i<=n;i++){
            if((k&(1<<(i-1)))==0)
                continue;
            for(j=1;j<=n;j++){
                if(i==j)
                    continue;
                if((k&(1<<(i-1)))==0)
                    continue;
                dp[i][k]=min(dp[i][k],dp[j][k-(1<<(i-1))]+a[i][j]);
            }
        }
    }
    for(i=1;i<=n;i++){
        ans=min(ans,dp[i][(1<<n)-1]);
    }
    printf("%.2f\n",ans);
    return 0;
}

7. [SCOI 2005] 互不侵犯

传送门

#include <algorithm>
#include <iostream>
using namespace std;
long long sta[2005], sit[2005], f[15][2005][105];
int n, k, cnt;
void dfs(int x, int num, int cur) {
    if (cur >= n) {  // 有新的合法状态
        sit[++cnt] = x;
        sta[cnt] = num;
        return;
    }
    dfs(x, num, cur + 1);  // cur位置不放国王
    dfs(x + (1 << cur), num + 1,
        cur + 2);  // cur位置放国王，与它相邻的位置不能再放国王
}
bool compatible(int j, int x) {
    if (sit[j] & sit[x]) return false;
    if ((sit[j] << 1) & sit[x]) return false;
    if (sit[j] & (sit[x] << 1)) return false;
    return true;
}
int main() {
    cin >> n >> k;
    dfs(0, 0, 0);  // 先预处理一行的所有合法状态
    for (int j = 1; j <= cnt; j++) f[1][j][sta[j]] = 1;
    for (int i = 2; i <= n; i++)
        for (int j = 1; j <= cnt; j++)
            for (int x = 1; x <= cnt; x++) {
                if (!compatible(j, x)) continue;  // 排除不合法转移
                for (int l = sta[j]; l <= k; l++) f[i][j][l] += f[i - 1][x][l - sta[j]];
            }
    long long ans = 0;
    for (int i = 1; i <= cnt; i++) ans += f[n][i][k];  // 累加答案
    cout << ans << endl;
    return 0;
}