题目大意:
给你n个双元组 < a i , b i > <a_i,b_i> <ai,bi>,可以选取i,j产生四元组 < a i , b i , a j , b j > <a_i,b_i,a_j,b_j> <ai,bi,aj,bj>,求可以使得四元组的LIS长度为1,2,3,4的 < i , j > <i,j> <i,j>个数(i和j可以相等)。
解题思路:
lis为4和为1的情况容易考虑:
LIS为4时:(ai < bi && aj<bj && bi < aj)
LIS为1时:(ai >= bi && aj >= bj && bi >= aj)
现在考虑LIS为3的情况:
- 当aj<bj时:
当ai < bi: aj<=bi<bj时,一定是3,bi>=bj时,ai<aj的情况下是3,其他情况不是
当ai >= bi, bi < aj时一定是3,其他情况不是 - 当aj >= bj时:
只有ai<bi且bi<aj的时候是3.
得到1,3,4的答案后,2的答案就是总数n*n减去1,3,4的答案.
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 1e5 + 50;
ll up[1050], down[1050];
ll d[1050][1050];
int n, m;
int x[maxn], y[maxn];
ll ans[5];
int main(){
cin>>n>>m;
for(int i = 1; i <= n; ++i){
scanf("%d%d", &x[i], &y[i]);
if(x[i] < y[i]) up[y[i]]++;
else down[y[i]]++;
d[x[i]][y[i]]++;
}
for(int i = 1; i <= m; ++i){
up[i] += up[i-1];
down[i] += down[i-1];
for(int j = 1; j <= m; ++j) d[i][j] += d[i-1][j] + d[i][j-1] - d[i-1][j-1];
}
for(int i = 1; i <= n; ++i){
if(y[i] > x[i]){
ans[4] += up[x[i]-1];
ans[3] += up[y[i]-1] - up[x[i]-1] + down[x[i]-1];
ans[3] += d[x[i]-1][m] - d[x[i]-1][y[i]-1];
}
else{
ans[1] += down[m]-down[x[i]-1];
ans[3] += up[x[i]-1];
}
}
ll sum = (ll)n*n;
ans[2] = sum-ans[1]-ans[3]-ans[4];
for(int i = 1; i <= 4; ++i){
if(i > 1) printf(" ");
printf("%lld", ans[i]);
}
}