Wannafly挑戰賽26-C 七彩線段
阿新 • • 發佈:2019-02-06
思路:狀態DP,首先對線段按照右端點從小到大排序,dp[i][j]:表示到第i條線段時狀態為j的最大長度。
那麼狀態轉移方程為 dp[i][j|a[i].p]=max(dp[i][j|a[i].p|,dp[k][j]+a[i].s) (a[i].p為顏色狀態,a[i].s為第i條線段長度,表示取當前線段時的最大長度,k為與當前線段不接觸的最右邊的線段)
dp[i][j]=max(dp[i][j],dp[i-1][j]);
Code :
#include<iostream> #include<algorithm> using namespace std; const int MAX_N=100005; const int MAX_M=130; struct node{ int l,r; int p; int s; bool operator<(const node &p)const{ return r<p.r; } }; int n,m; int dp[MAX_N][MAX_M]; node a[MAX_N]; int main() { ios::sync_with_stdio(false); cin>>n>>m; for(int i=1;i<=n;++i) { cin>>a[i].l>>a[i].r>>a[i].p; a[i].p=1<<(a[i].p-1); a[i].s=a[i].r-a[i].l; } sort(a+1,a+n+1); for(int i=1;i<=n;++i) dp[i][a[i].p]=a[i].s; int mm=(1<<7)-1; for(int i=1;i<=n;++i) { int k=lower_bound(a,a+n,node{0,a[i].l,0})-a-1; for(int j=0;j<=mm;++j) { dp[i][j]=max(dp[i][j],dp[i-1][j]); int t=j|a[i].p; if(dp[k][j]) dp[i][t]=max(dp[i][t],dp[k][j]+a[i].s); } } int ans=0; for(int i=0;i<=mm;++i) { int s=0,t=i; while(t){ if(t&1) ++s; t>>=1; } if(s==m) ans=max(ans,dp[n][i]); } if(!ans) ans=-1; cout<<ans<<endl; return 0; }