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<<30
using namespace std;
/*
用扫描法,记录l和r
10分 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;
}