您的位置:首页 > 其它

BestCoder 2nd Anniversary B hdu 5719 树状数组

2016-07-17 23:17 75 查看
链接:戳这里

Arrange  

 Time Limit: 8000/4000 MS (Java/Others)   Memory Limit: 262144/262144 K (Java/Others)
问题描述

Cupid一不小心将爱情之箭射到了自己,他爱上了Psyche。

这引起了他的母亲Venus的注意。Venus将Psyche带到了一堆打乱的谷堆旁。

这儿共有 n n堆稻谷,编号为  1到  n。Psyche需要将这些谷堆以某种顺序排列,设最终排在第 ii位的谷堆是 A​i。

她得知了一些该排列的要求:

  1. 对于任意整数i∈[1,n], A1 ,A​2 ,...,A​i​​ 的最小值为 B​i​​ 。

  2. 对于任意整数i∈[1,n], A1 ,A​2 ,...,Ai​​ 的最大值为 Ci 。

现在Psyche想知道,共有多少种合法的排列。由于答案可能很大,输出时对 998244353 取模。

输入描述

第一行,一个整数 T  (1≤T≤15),代表数据组数。

对于每组数据,第一行有一个整数 n (1≤n≤10e5),代表排列大小。

第二行, n n个整数,第 i i个整数为 B​i​​ (1≤B​i​​ ≤n)。

第三行, n n个整数,第 i i个整数为 C​i (1≤C​i​​ ≤n)。

输出描述

输出 T 行,对于每组数据输出答案对 998244353取模的结果。

输入样例

2

3

2 1 1

2 2 3

5

5 4 3 2 1

1 2 3 4 5

输出样例

1

0

Hint

对于第一组数据,只有一种合法的排列 (2,1,3)。

对于第二组数据,没有合法的排列。

思路:

只有当Bi和Ci都没有变化的时候才计算贡献,那么剩下的是统计区间[Bi,Ci]里还有多少个数能选

用树状数组计算区间和,每加入一个数对应位置上+1。当Bi和Ci都没变得时候,统计完贡献以后也要插入一个数。

这个数可以是Bi(区间[mn,mx]两边一定在慢慢靠近,所以选Bi每次都会统计进去)。

我们只需要计算区间上还有多少个数没有放,不一定要确定到哪些数

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<vector>
#include <ctime>
#include<queue>
#include<set>
#include<map>
#include<stack>
#include<iomanip>
#include<cmath>
#define mst(ss,b) memset((ss),(b),sizeof(ss))
#define maxn 0x3f3f3f3f
#define MAX 1000100
///#pragma comment(linker, "/STACK:102400000,102400000")
typedef long long ll;
typedef unsigned long long ull;
#define INF (1ll<<60)-1
#define mod 998244353
using namespace std;
int n;
int B[100100],C[100100];
int num[100100];
int lowbit(int x){
return x&-x;
}
void update(int x){
while(x<=n){
num[x]++;
x+=lowbit(x);
}
}
int getsum(int x){
int ans=0;
while(x>=1){
ans+=num[x];
x-=lowbit(x);
}
return ans;
}
int main(){
int T;
scanf("%d",&T);
while(T--){
scanf("%d",&n);
mst(num,0);
int flag=0;
for(int i=1;i<=n;i++) {
scanf("%d",&B[i]);
if(i!=1 && B[i]>B[i-1]) flag=1;
}
for(int i=1;i<=n;i++) {
scanf("%d",&C[i]);
if(B[i]>C[i]) flag=1;
if(C[i]<C[i-1]) flag=1;
}
ll ans=1;
int mn=B[1],mx=C[1];
update(mn);
if(mn!=mx) flag=1;
for(int i=2;i<=n;i++){
if(B[i]!=mn && C[i]==mx){
mn=B[i];
update(mn);
} else if(B[i]==mn && C[i]!=mx){
mx=C[i];
update(mx);
} else if(B[i]!=mn && C[i]!=mx){
mn=B[i];
mx=C[i];
flag=1;
break;
} else if(B[i]==mn && C[i]==mx){
int x=mx-mn+1;
int y=getsum(mx)-getsum(mn-1);
if(x>y) update(mn);
ans*=max(0,(x-y));
}
ans%=mod;
}
if(flag) {
cout<<0<<endl;
continue;
}
printf("%I64d\n",ans);
}
return 0;
}
/*
10
5
5 1 1 1 1
5 5 5 5 5
5
2 2 2 2 1
2 5 5 5 5

*/
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: