1. 程式人生 > >Wannafly挑戰賽26-C 七彩線段

Wannafly挑戰賽26-C 七彩線段

思路:狀態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;
}