六年级的暑假,在腾讯实习的xiaoxin巨从leader那得到了很多西瓜糖。每个西瓜糖都有一个属性代表它的甜度值,用一个正整数表示。 xiaoxin从小就特别会玩,他先把所有糖果排成一条直线,每次从一个区间内选出三个连续的且甜度值不减的西瓜糖来吃,即: 如果他选择了三元组 (ai,aj,ak) 那么: 1. j=i+1,k=j+1 2. ai≤aj≤ak 好奇的xiaoxin想知道每次吃糖果,他有多少种吃法? 我们认为两个三元组 (a0,a1,a2),(b0,b1,b2) 是不同的,当且仅当: a0≠b0 or a1≠b1 or a2≠b2.
多组测试数据, 第一行为一个整数(T≤10)表示测试数据组数。 对于每组测试数据,第一行是一个整数 n(1≤n≤200,000) 表示西瓜糖的个数。接下来 n个数表示每个西瓜糖的甜度值(0≤ai≤1000,000,000)。接下来有 Q(1≤200,000) 行代表询问的次数,每个询问用两个整数 l,r(1≤l≤r≤n) 表示。
对于每次询问,输出一个数,表示这次xiaoxin的吃法的数目。
1 5 1 2 3 4 5 3 1 3 1 4 1 5
1 2 3
#include<stdio.h> #include<iostream> #include<string.h> #include<string> #include<ctype.h> #include<math.h> #include<set> #include<map> #include<vector> #include<queue> #include<bitset> #include<algorithm> #include<time.h> using namespace std; void fre() { freopen("c://test//input.in", "r", stdin); freopen("c://test//output.out", "w", stdout); } #define MS(x,y) memset(x,y,sizeof(x)) #define MC(x,y) memcpy(x,y,sizeof(x)) #define MP(x,y) make_pair(x,y) #define ls o<<1 #define rs o<<1|1 typedef long long LL; typedef unsigned long long UL; typedef unsigned int UI; template <class T1, class T2>inline void gmax(T1 &a, T2 b) { if (b>a)a = b; } template <class T1, class T2>inline void gmin(T1 &a, T2 b) { if (b<a)a = b; } const int N = 2e5 + 10, M = 0, Z = 1e9 + 7, ms63 = 0x3f3f3f3f; int casenum, casei; int n, m; int c[N]; //用于记录原数的数组 int b[N]; //树状数组,我们利用其来统计在某个范围内有多少个数字的前驱<l int nxt[N]; //nxt[i]表示与c[i]权值相同的下一个数的位置 int l[N], r[N], o[N]; //用于处理询问的数组 int ans[N]; //用于输出答案的数组 map< pair<int, pair<int, int> >, int>mop; map< pair<int, pair<int, int> >, int>::iterator it; bool cmp(int x, int y) { return l[x] < l[y]; } int check(int x) { int ret = 0; for (; x; x -= x&-x)ret += b[x]; return ret; } void add(int x, int val) { for (; x <= n; x += x&-x)b[x] += val; } int main() { scanf("%d", &casenum); for (casei = 1; casei <= casenum; ++casei) { mop.clear(); scanf("%d", &n); for (int i = 1; i <= n; ++i)b[i] = nxt[i] = 0; for (int i = 1; i <= n; ++i)scanf("%d", &c[i]); for (int i = 3; i <= n; ++i) { if (i >= 3 && c[i - 2] <= c[i - 1] && c[i - 1] <= c[i]) { int x = mop[MP(c[i - 2], MP(c[i - 1], c[i]))]; if (x == 0)add(i, 1); else nxt[x] = i; mop[MP(c[i - 2], MP(c[i - 1], c[i]))] = i; } } scanf("%d", &m); for (int i = 1; i <= m; ++i)scanf("%d%d", &l[i], &r[i]), o[i] = i; sort(o + 1, o + m + 1, cmp); int p = 1; for (int i = 1; i <= m; ++i) { int lft = l[o[i]]; int rgt = r[o[i]]; while (p <= lft + 1) { if(nxt[p])add(nxt[p], 1); ++p; } if (rgt - lft >= 2)ans[o[i]] = check(rgt) - check(lft + 1); else ans[o[i]] = 0; } for (int i = 1; i <= m; ++i)printf("%d\n", ans[i]); } return 0; } /* 【trick&&吐槽】 1,注意,我们一定要注意我们对于一个[l,r]的询问区间,其所对应的真实区间其实是[l+2,r] 2,边界问题把握好,比如—— if (rgt - lft >= 2)ans[o[i]] = check(rgt) - check(lft + 1); else ans[o[i]] = 0; 这第二句不能忽视 3,复杂度固然重要,但是代码复杂度也非常重要,否则实现不了程序再快又有什么用 【题意】 问你一个区间中有多少个本质不同的连续不下降三元组 即(ai,aj,ak)满足i+1=j,j+1=k且a[i]<=a[j]<=a[k] 【类型】 树状数组 排序 【分析】 我们要统计的是区间范围[l,r]内有多少个三元组的pre小于l。 为了实现这个功能,一个好的做法是,在数据结构中,只维护所有pre<l的数。 显然,随着查询区间左界l的增大,pre<l的数是增多的。 于是,我们队所有的查询按照左界做排序。 我们之前会把pre<l的所有数都放进位置恰好其下标所对应的数据结构(如树状数组)中。 那我们查询的时候,只要查询[lft+2,rgt]期间有多少个数即可 【时间复杂度&&优化】 O(nlogn) */