1. 程式人生 > >2019年第十屆藍橋杯省賽-糖果(一維狀壓dp)

2019年第十屆藍橋杯省賽-糖果(一維狀壓dp)

spa arr oid scan static main println sta 枚舉

技術分享圖片

技術分享圖片

看到20的數據量很容易想到狀壓dp。

開1<<20大小的數組來記錄狀態,枚舉n個糖包,將其放入不同狀態中(類似01背包思想)

時間復雜度O(n*(2^20))。

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    static Scanner sc = new Scanner(System.in);
    static int[][] a = new int[105][25];
    static int[] sta = new int[105];
    
static int[] dp = new int[(1<<20)+5]; public static void main(String[] args) { int n=sc.nextInt(); int m=sc.nextInt(); int k=sc.nextInt(); Arrays.fill(dp, -1); dp[0]=0; for(int i=1;i<=n;i++){ for(int j=1;j<=k;j++){ a[i][j]
=sc.nextInt(); sta[i]|=1<<(a[i][j]-1); } dp[sta[i]]=1; } for(int i=1;i<=n;i++){ for(int j=0;j<(1<<m);j++){ if(dp[j]==-1) continue; if(dp[j|sta[i]]==-1||dp[j]+1<dp[j|sta[i]]){ dp[j
|sta[i]]=dp[j]+1; } } } System.out.println(dp[(1<<m)-1]); } }

2019年第十屆藍橋杯省賽-糖果(一維狀壓dp)