Wannafly挑战赛26-C 七彩线段

地址：https://www.nowcoder.com/acm/contest/212/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;
}