3262: 陌上花开
Time Limit: 20 Sec Memory Limit: 256 MBSubmit: 217 Solved: 104
[Submit][Status]
Description
有n朵花,每朵花有三个属性:花形(s)、颜色(c)、气味(m),又三个整数表示。现要对每朵花评级,一朵花的级别是它拥有的美丽能超过的花的数量。定义一朵花A比另一朵花B要美丽,当且仅当Sa>=Sb,Ca>=Cb,Ma>=Mb。显然,两朵花可能有同样的属性。需要统计出评出每个等级的花的数量。
Input
第一行为N,K (1 <= N <= 100,000, 1 <= K <= 200,000 ), 分别表示花的数量和最大属性值。
以下N行,每行三个整数si, ci, mi (1 <= si, ci, mi <= K),表示第i朵花的属性
Output
包含N行,分别表示评级为0...N-1的每级花的数量。
Sample Input
10 3
3 3 3
2 3 3
2 3 1
3 1 1
3 1 2
1 3 1
1 1 2
1 2 2
1 3 2
1 2 1
3 3 3
2 3 3
2 3 1
3 1 1
3 1 2
1 3 1
1 1 2
1 2 2
1 3 2
1 2 1
Sample Output
3
1
3
0
1
0
1
0
0
1
1
3
0
1
0
1
0
0
1
HINT
1 <= N <= 100,000, 1 <= K <= 200,000
Source
题解:
三维。。。第一维排序,第二维树状数组,第三维treap
具体实现就是每个树状数组的节点保存一颗treap,然后就可以查询了。
树套树就是强大!
刚开始为了节省时间把相同的节点同时加入,然后发现这样的rank就变小了T_T。。。
唉?貌似也可以搞?
代码:
1 #include<cstdio> 2 #include<cstdlib> 3 #include<cmath> 4 #include<cstring> 5 #include<algorithm> 6 #include<iostream> 7 #include<vector> 8 #include<map> 9 #include<set> 10 #include<queue> 11 #include<string> 12 #define inf 1000000000 13 #define maxn 200000+1000 14 #define maxm 4000000 15 #define eps 1e-10 16 #define ll long long 17 #define pa pair<int,int> 18 #define for0(i,n) for(int i=0;i<=(n);i++) 19 #define for1(i,n) for(int i=1;i<=(n);i++) 20 #define for2(i,x,y) for(int i=(x);i<=(y);i++) 21 #define for3(i,x,y) for(int i=(x);i>=(y);i--) 22 #define mod 1000000007 23 using namespace std; 24 inline int read() 25 { 26 int x=0,f=1;char ch=getchar(); 27 while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} 28 while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();} 29 return x*f; 30 } 31 struct rec{int x,y,z;}a[maxn]; 32 int n,m,tot,ls[maxm],rs[maxm],s[maxm],w[maxm],rnd[maxm],v[maxm],sum[maxn],ans[maxn],root[maxn]; 33 inline bool cmp(rec a,rec b) 34 { 35 return a.x<b.x||(a.x==b.x&&a.y<b.y)||(a.x==b.x&&a.y==b.y&&a.z<b.z); 36 } 37 inline void pushup(int k) 38 { 39 s[k]=s[ls[k]]+s[rs[k]]+w[k]; 40 } 41 inline void rturn(int &k) 42 { 43 int t=ls[k];ls[k]=rs[t];rs[t]=k;pushup(k);pushup(t);k=t; 44 } 45 inline void lturn(int &k) 46 { 47 int t=rs[k];rs[k]=ls[t];ls[t]=k;pushup(k);pushup(t);k=t; 48 } 49 void ins(int &k,int x) 50 { 51 if(!k){k=++tot;s[k]=w[k]=1;ls[k]=rs[k]=0;rnd[k]=rand();v[k]=x;return;} 52 s[k]++; 53 if(x==v[k]){w[k]++;return;} 54 else if(x<v[k]){ins(ls[k],x);if(rnd[ls[k]]>rnd[k])rturn(k);} 55 else {ins(rs[k],x);if(rnd[rs[k]]>rnd[k])lturn(k);}; 56 } 57 int rank(int k,int x) 58 { 59 if(!k)return 0; 60 if(x==v[k])return s[ls[k]]+w[k]; 61 else if(x<v[k])return rank(ls[k],x); 62 else return s[ls[k]]+w[k]+rank(rs[k],x); 63 } 64 inline void insert(int x,int y) 65 { 66 for(;x<=m;x+=x&(-x))ins(root[x],y); 67 } 68 inline int query(int x,int y) 69 { 70 int t=0; 71 for(;x;x-=x&(-x))t+=rank(root[x],y); 72 return t; 73 } 74 int main() 75 { 76 freopen("input.txt","r",stdin); 77 freopen("output.txt","w",stdout); 78 n=read();m=read(); 79 for1(i,n)a[i].x=read(),a[i].y=read(),a[i].z=read(); 80 sort(a+1,a+n+1,cmp); 81 for1(i,n)sum[i]=1; 82 for1(i,n) 83 { 84 if(a[i].x==a[i+1].x&&a[i].y==a[i+1].y&&a[i].z==a[i+1].z)sum[i+1]+=sum[i]; 85 else ans[query(a[i].y,a[i].z)]+=sum[i]; 86 insert(a[i].y,a[i].z); 87 } 88 for0(i,n-1)printf("%d ",ans[i]); 89 return 0; 90 }
我擦,一块加也行,那么rank就等于 query+sum[i]-1
结果加了这个优化才快了几百毫秒 T_T
代码:
1 #include<cstdio> 2 #include<cstdlib> 3 #include<cmath> 4 #include<cstring> 5 #include<algorithm> 6 #include<iostream> 7 #include<vector> 8 #include<map> 9 #include<set> 10 #include<queue> 11 #include<string> 12 #define inf 1000000000 13 #define maxn 200000+1000 14 #define maxm 4000000 15 #define eps 1e-10 16 #define ll long long 17 #define pa pair<int,int> 18 #define for0(i,n) for(int i=0;i<=(n);i++) 19 #define for1(i,n) for(int i=1;i<=(n);i++) 20 #define for2(i,x,y) for(int i=(x);i<=(y);i++) 21 #define for3(i,x,y) for(int i=(x);i>=(y);i--) 22 #define mod 1000000007 23 using namespace std; 24 inline int read() 25 { 26 int x=0,f=1;char ch=getchar(); 27 while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} 28 while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();} 29 return x*f; 30 } 31 struct rec{int x,y,z;}a[maxn]; 32 int n,m,tot,ls[maxm],rs[maxm],s[maxm],w[maxm],rnd[maxm],v[maxm],sum[maxn],ans[maxn],root[maxn]; 33 inline bool cmp(rec a,rec b) 34 { 35 return a.x<b.x||(a.x==b.x&&a.y<b.y)||(a.x==b.x&&a.y==b.y&&a.z<b.z); 36 } 37 inline void pushup(int k) 38 { 39 s[k]=s[ls[k]]+s[rs[k]]+w[k]; 40 } 41 inline void rturn(int &k) 42 { 43 int t=ls[k];ls[k]=rs[t];rs[t]=k;pushup(k);pushup(t);k=t; 44 } 45 inline void lturn(int &k) 46 { 47 int t=rs[k];rs[k]=ls[t];ls[t]=k;pushup(k);pushup(t);k=t; 48 } 49 void ins(int &k,int x,int cnt) 50 { 51 if(!k){k=++tot;s[k]=w[k]=cnt;ls[k]=rs[k]=0;rnd[k]=rand();v[k]=x;return;} 52 s[k]+=cnt; 53 if(x==v[k]){w[k]+=cnt;return;} 54 else if(x<v[k]){ins(ls[k],x,cnt);if(rnd[ls[k]]>rnd[k])rturn(k);} 55 else {ins(rs[k],x,cnt);if(rnd[rs[k]]>rnd[k])lturn(k);}; 56 } 57 int rank(int k,int x) 58 { 59 if(!k)return 0; 60 if(x==v[k])return s[ls[k]]+w[k]; 61 else if(x<v[k])return rank(ls[k],x); 62 else return s[ls[k]]+w[k]+rank(rs[k],x); 63 } 64 inline void insert(int x,int y,int cnt) 65 { 66 for(;x<=m;x+=x&(-x))ins(root[x],y,cnt); 67 } 68 inline int query(int x,int y) 69 { 70 int t=0; 71 for(;x;x-=x&(-x))t+=rank(root[x],y); 72 return t; 73 } 74 int main() 75 { 76 freopen("input.txt","r",stdin); 77 freopen("output.txt","w",stdout); 78 n=read();m=read(); 79 for1(i,n)a[i].x=read(),a[i].y=read(),a[i].z=read(); 80 sort(a+1,a+n+1,cmp); 81 for1(i,n)sum[i]=1; 82 for1(i,n) 83 if(a[i].x==a[i+1].x&&a[i].y==a[i+1].y&&a[i].z==a[i+1].z)sum[i+1]+=sum[i]; 84 else 85 { 86 ans[query(a[i].y,a[i].z)+sum[i]-1]+=sum[i]; 87 insert(a[i].y,a[i].z,sum[i]); 88 } 89 for0(i,n-1)printf("%d ",ans[i]); 90 return 0; 91 }