七彩线段

题目链接
听说彩虹有七种颜色？
一维坐标轴上n条线段，每条线段左端点l，右端点r，颜色为c，从中选m种颜色的互不接触的线段，每种颜色可选多条，所选线段的总长度最长为多少？
输入描述:
第一行2个整数 n, m;
接下来n行，每行3个整数l, r, c。
输出描述:
一个整数，表示所选线段的最长的总长度；若选不了，输出-1；
示例1
输入
复制
4 2
1 3 1
4 5 1
5 8 2
7 9 3
输出
复制
5
示例2
输入
复制
4 3
1 3 1
4 5 1
5 8 2
7 9 3
输出
复制
-1

思路分析：
首先定义一个结构体存相关数据，由于题目要求所选线段不能相关接触，故后面需要二分查找线段，所以我们安装每天线段的末值从小到达排序：

struct Node{
	int l;
	int r;
	int c;
	friend bool operator <(const Node& a,const Node& b){
		return a.r<b.r;
	}
}a[maxn];

接着我们需要定义一个数组，dp[maxn][1<<7];
第一维代表线段，第二维代表选取的颜色，其中选取的颜色用二进制表示
由于选取的颜色用二进制表示，故我们可以预处理每个数组代表选取了多少种颜色

	int numbit[1<<7];
	for(int i=1;i<(1<<7);i++){
		numbit[i]=numbit[i&(i-1)]+1;
	}

dp[i][j]便代表前i条线段在选取颜色情况为j的时候的最大长度。
首先初始化全部初始化为-1,dp[0][0]=0;

	memset(dp,-1,sizeof(dp));//首先初始化 
	dp[0][0]=0;

首先对与第i个线段，为了符合题目中的不相接触
我们二分查找距离i号线段最近的不接触线段

		int l=0;
		int r=i-1;
		int def=0;
		while(l<=r){
			int mid=(l+r)>>1;
			if(a[mid].r<a[i].l){
				def=mid;
				l=mid+1;
			}
			else r=mid-1;
		}

这时候我们已经找到了最近不接触线段了，开始状态转移
首先考虑不选该条线段，如果dp[i-1][j]!=-1，可以直接与dp[i][j]=max(dp[i-1][j],dp[i][j]);
再考虑选该线段的情况，若干dp[def][j]!=-1，则可以由这转移过来
dp[i][j|(1<<(a[i].c-1))]=max(dp[i][j|(1<<(a[i].c-1))],dp[def][j]+a[i].r-a[i].l);
这里注意要跟新第二维的选取线段颜色
最后判断是否该情况下选取了m种颜色，若选取了跟新最大值即可

		for(int j=0;j<(1<<7);j++){
			dp[i][j]=max(dp[i][j],dp[i-1][j]);//不选第i个的情况：利用前面的更新 
			if(dp[def][j]!=-1){
				dp[i][j|(1<<(a[i].c-1))]=max(dp[i][j|(1<<(a[i].c-1))],dp[def][j]+a[i].r-a[i].l);
			}
			if(numbit[j|(1<<(a[i].c-1))]==m) ans=max(ans,dp[i][j|(1<<a[i].c-1)]);
		}

完整代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
int n,m;
ll dp[maxn][1<<7];
struct Node{
	int l;
	int r;
	int c;
	friend bool operator <(const Node& a,const Node& b){
		return a.r<b.r;
	}
}a[maxn];
int numbit[1<<7];
int main() 
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	for(int i=1;i<(1<<7);i++){
		numbit[i]=numbit[i&(i-1)]+1;
	}
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i].l>>a[i].r>>a[i].c;
	}
	sort(a+1,a+1+n);
	memset(dp,-1,sizeof(dp));//首先初始化 
	dp[0][0]=0;
	ll ans=-1;
	for(int i=1;i<=n;i++){
		int l=0;
		int r=i-1;
		int def=0;
		while(l<=r){
			int mid=(l+r)>>1;
			if(a[mid].r<a[i].l){
				def=mid;
				l=mid+1;
			}
			else r=mid-1;
		}
		for(int j=0;j<(1<<7);j++){
			dp[i][j]=max(dp[i][j],dp[i-1][j]);//不选第i个的情况：利用前面的更新 
			if(dp[def][j]!=-1){
				dp[i][j|(1<<(a[i].c-1))]=max(dp[i][j|(1<<(a[i].c-1))],dp[def][j]+a[i].r-a[i].l);
			}
			if(numbit[j|(1<<(a[i].c-1))]==m) ans=max(ans,dp[i][j|(1<<a[i].c-1)]);
		}
		
	}
	cout<<ans<<endl;
}