Atcoder AGC012F : Prefix Median

18313 단어 DP 및 DP 최적화
전송문
문제풀이: 먼저 a a 정렬을 하고 b b b 정렬을 구성할 수 있으면:
  • a i ≤ b i ≤ a 2 n − i a_i\le b_i\le a_{2n-i} ai​≤bi​≤a2n−i​.

  • 4
  • i, j i, j, i < j, b j < b i < b j + 1 i\lt j, b 없음j\lt b_i\lt b_{j+1} i
    b b b b가 뒤로 갈수록 제한이 강해지는 것을 관찰했습니다. 우리는 뒤에서 앞으로 DP, 유지 보수 f i, j, k f{i, j, k} fi, j, k는 ii i개를 나타내고 선택 수량은 bi+1b 보다 작다{i+1}bi+1은 jj개가 있고, bi+1b 보다 크다.{i+1}bi+1은 kk개가 있습니다.그 다음에 뭘 쓰는지 하나하나 열거하면 돼.시간 복잡도 O(n4)O(n^4)O(n4).
    #include 
    using namespace std;
    
    const int RLEN=1<<18|1;
    inline char nc() {
    	static char ibuf[RLEN],*ib,*ob;
    	(ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
    	return (ib==ob) ? -1 : *ib++;
    }
    inline int rd() {
    	char ch=nc(); int i=0,f=1;
    	while(!isdigit(ch)) {if(ch=='-')f=-1; ch=nc();}
    	while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();}
    	return i*f;
    }
    
    const int N=105, mod=1e9+7;
    inline void add(int &x,int y) {x=(x+y>=mod) ? (x+y-mod) : (x+y);}
    
    int n,a[N];
    int f[N/2][N][N];
    
    int main() {
    	n=rd();
    	for(int i=1;i<=2*n-1;i++) a[i]=rd();
    	sort(a+1,a+2*n);
    	
    	f[n][1][0]=1;
    	for(int i=n-1;i>=1;i--) {
    		int al=(a[i]!=a[i+1]), ar=(a[2*n-i]!=a[2*n-i-1]);
    		for(int l=0;l<=2*n-1;++l)
    			for(int r=0;r+l<=2*n-1;++r) if(f[i+1][l][r]) {
    				int &t=f[i+1][l][r];	
    				for(int dl=1;dl<=l+al;++dl) add(f[i][l+al-dl+1][r+ar+(dl>1)],t);
    				for(int dr=1;dr<=r+ar;++dr) add(f[i][l+al+1][r+ar-dr],t);
    			}
    	}
    	
    	int ans=0;
    	for(int l=0;l<=2*n-1;++l)
    		for(int r=0;r+l<=2*n-1;++r)
    			add(ans,f[1][l][r]);
    	cout<<ans<<'
    '
    ; }
  • 좋은 웹페이지 즐겨찾기