NOIP 2011 选择客栈

最近学的高效算法设计起到了用场，第一次用对拍来确保正确性

首先写好暴力

本题我的解法是扫描

left[x]为当前位置左边颜色k的个数

right[x]当前位置右边k的个数

读入时计算right数组

从第一个客栈开始扫描

把这个客栈当成中间点，然后对每种颜色计算left[i]*right[i]，顺便维护left和right

解决了？NO

有一个很大的问题：会有很多重复，两个隔很远的客栈会被算很多次

所以每次都要把left清零(前提条件是这个中间点能当作客栈)

还有一个问题，枚举颜色时会枚举到当前位置的颜色，比较复杂，单独处理

之前计算的都是跨越中间点的情况，还有不跨越的，这就是当前颜色的麻烦的地方

要分类讨论之类的

我做完这题之后想通了，其实答案是left和right的乘积-1不是巧合

进去的时候left++了，-1减去的就是他 也就是这个点既是left又是right的情况

#include <iostream>#include <cstdio>#include <cstring>#include <string>#include <ctime>#include <cmath>#include <vector>#include <deque>#include <algorithm>#include <list>#include <map>#include <set>#include <queue>#include <stack>#define REP(i,n) for(int i=0;i<(n);i++)#define REP1(i,n) for(int i=1;i<=(n);i++)#define FOR(i,a,b) for (int i=(a);i<=(b);i++)#define CLR(x,n) memset(x,n,sizeof(x))#define PN printf("\n")#define read(x) scanf("%d",&x)#define read2(x,y) scanf("%d%d",&x,&y)#define read3(x,y,z) scanf("%d%d%d",&x,&y,&z)#define write(x) printf("%d",x)#define write1(x) printf("%d ",x)#define writeln(x) printf("%d\n",x)#define write2(x,y) printf("%d %d",x,y)#define writeln2(x,y) printf("%d %d\n",x,y)#define write3(x,y,z) printf("%d %d %d",x,y,z)#define writeln3(x,y,z) printf("%d %d %d\n",x,y,z)#define delta 1e-5#define INF 1<<30using namespace std;/*用扫描法,记录l和r10分 k和pr搞错 ..出现了一个很麻烦的情况 会少解幸好思想正确 修改一个式子就AC了 */ int n,k,pr;int l[100],r[100],a[200010],p[200010];int main(){    read3(n,k,pr);int ans=0;CLR(l,0);//writeln3(n,k,pr);    REP1(i,n){        read2(a[i],p[i]);//writeln2(a[i],p[i]);        r[a[i]]++;    }    REP1(i,n){    l[a[i]]++;        if (p[i]<=pr){        REP(j,k)if (j!=a[i]){ //原来写的居然是pr         ans+=l[j]*r[j];//printf("%d %d %d\n",j,l[j],r[j]);l[j]=0;}ans+= l[a[i]] * r[a[i]] -1;//经过修改 //printf(" %d %d\n",l[a[i]],r[a[i]]); l[a[i]]=0;          }        r[a[i]]--;     }    writeln(ans);    //system("pause");    return 0;}