洛谷1311 选择客栈

NOIP2011 Day1 T2

原题地址

看上去是个DP然而可以拿好几种算法跑过。。。结合数据范围，O(n2)的时间复杂度只能过50%的点，要想AC，正解时间复杂度应该远小于O(n2)。看题解有大犇拿线段树A过了，然而个人觉得这个题用线段树有点大材小用，拿O(nk)的解法完全可过。

解题思路

一言以蔽之，从前向后暴力找。

观察可得显然要把客栈分成k类，对于每一类求可行方案数。一提到方案数，第一反应就是DP，事实上它确实满足DP的条件，最优子结构+无后效性：对于每一类的第i个咖啡馆，如果能与前一个配对，那么一定与前i-1个配对都能配对，如果不能，那么就保持前一个的状态。这一关系不受之后咖啡馆的影响。但是转移的时候要注意不同情况分类讨论，受当前咖啡馆自身情报（信息）的影响会有不同的转移，具体情况参考代码。

参考代码

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
using namespace std;
long long ans=0;
int a[200005],b[200005];
int st[55],ed[55],f[55];
long long dp[4005];
int main()
{
	int num=0;
	int n,k,p;
	cin>>n>>k>>p;
	for (int i=1;i<=n;i++)
	{
		scanf("%d%d",&a[i],&b[i]);
		if(!f[a[i]]) st[a[i]]=i;  //记录第a[i]类咖啡馆第一次出现的位置，减少循环搜索次数；
		else ed[a[i]]=i;  //记录第a[i]类咖啡馆最后一次出现的位置；
		f[a[i]]++;  //第a[i]类的个数；
    }
    for (int i=0;i<k;i++)  //枚举每类咖啡馆
    {
    	if (f[i]<2) continue;  //不足两个，直接跳过；
    	int l=st[i],r=ed[i];
    	bool ff=0;  //用来判断相邻两家店能否配对；
    	int cnt=1;  //记录当前是第几家店；
    	memset(dp,0,sizeof(dp));
    	if (b[l]<=p) ff=1; else ff=0;  //自身满足条件，后面的一定能配对，否则向后找找再说；
		for (int j=l+1;j<=r;j++)
    	{
    		if (a[j]==i)  //是第i类咖啡馆；
    		{
    			cnt++;
    			if (b[j]<=p)  //自身满足条件，一定能和前面的配对；
    			{
    				dp[cnt]=cnt-1; 
    				ff=true;  //自身满足条件，后面的一定能配对；
				}else  //自身不满足条件；
				{
					if (ff)  //前面（到前一个咖啡店之间）有满足消费<=p的点，一定能和前一个配对；
					{
						dp[cnt]=cnt-1;
						ff=false;  
					}else  //不能和前一个配对；
					{
						dp[cnt]=dp[cnt-1];
						ff=false;
					}
				}
				ans+=dp[cnt];  //累加答案；
			}else  //不是第i类，仅提供条件判断；
			{
				if (ff) continue;
				else
				{
					if (b[j]<=p) ff=true;
					continue;
				}
			}
		}
	}
	cout<<ans;
	return 0;
}