题目描述
给出一维线段长度n,客栈装修颜色k,咖啡店最高的花费p。
再给出n个点的装修颜色和咖啡店花费,问你在装修颜色相同的两家店,左右闭区间的情况下,区间咖啡店最小值小于等于p的集合有几个?
Solution
因为咖啡店是顺序给出的,我们直接对着先录取颜色和咖啡店花费,如果当前点的花费小于等于p,那么和当前节点相同颜色的在它左边的咖啡店都可以去了。
按着这个思想编程就可以了,还有一个注意的就是它左边的咖啡店不需要开个for去遍历了,运用pre前驱数组,判断可以去的最右边的点有没有大于等于当前颜色的前一个出现位置,如果大于了说明左边出现的同颜色点都可以选择。吧答案累加上去即可。
#pragma GCC target("avx,sse2,sse3,sse4,popcnt") #pragma GCC optimize("O2,O3,Ofast,inline,unroll-all-loops,-ffast-math") #include <bits/stdc++.h> using namespace std; #define js ios::sync_with_stdio(false);cin.tie(0); cout.tie(0) #define all(__vv__) (__vv__).begin(), (__vv__).end() #define endl "\n" #define pai pair<int, int> #define ms(__x__,__val__) memset(__x__, __val__, sizeof(__x__)) typedef long long ll; typedef unsigned long long ull; typedef long double ld; inline ll read() { ll s = 0, w = 1; char ch = getchar(); for (; !isdigit(ch); ch = getchar()) if (ch == '-') w = -1; for (; isdigit(ch); ch = getchar()) s = (s << 1) + (s << 3) + (ch ^ 48); return s * w; } inline void print(ll x, int op = 10) { if (!x) { putchar('0'); if (op) putchar(op); return; } char F[40]; ll tmp = x > 0 ? x : -x; if (x < 0)putchar('-'); int cnt = 0; while (tmp > 0) { F[cnt++] = tmp % 10 + '0'; tmp /= 10; } while (cnt > 0)putchar(F[--cnt]); if (op) putchar(op); } inline ll gcd(ll x, ll y) { return y ? gcd(y, x % y) : x; } ll qpow(ll a, ll b) { ll ans = 1; while (b) { if (b & 1) ans *= a; b >>= 1; a *= a; } return ans; } ll qpow(ll a, ll b, ll mod) { ll ans = 1; while (b) { if (b & 1)(ans *= a) %= mod; b >>= 1; (a *= a) %= mod; }return ans % mod; } inline int lowbit(int x) { return x & (-x); } const int dir[][2] = { {0,1},{1,0},{0,-1},{-1,0},{1,1},{1,-1},{-1,1},{-1,-1} }; const int MOD = 1e9 + 7; const int INF = 0x3f3f3f3f; const int N = 100 + 7; int tot[N], cnt[N], pre[N]; int main() { int n = read(), k = read(), p = read(); int ans = 0, r = 0; for (int i = 1; i <= n; ++i) { int a = read(), b = read(); if (b <= p) r = i; if (r >= pre[a]) tot[a] = cnt[a]; ans += tot[a]; pre[a] = i; ++cnt[a]; } print(ans); return 0; }