Submission #1150141

#	Time	Username	Problem	Language	Result	Execution time	Memory
1150141		nai0610	Skyscraper (JOI16_skyscraper)	C++20	15 / 100	1 ms	836 KiB

skyscraper

#include <bits/stdc++.h>
#define ll int64_t
#define ld long double
using namespace std;
// 
const int maxn =1e2+5;
const int mod = 1e9+7; // 998244353,1610612741
const ll inf = 1e18;
const ld pi = atan(1.0L)*4;
int n,K,a[maxn];
ll f[maxn][maxn][1005][3];
int main() {
    // freopen("../input.inp","r",stdin);
    // freopen("output.out","w",stdout);
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    clock_t start,end;
	start=clock();
	cin >>n>>K;
	for (int i=1;i<=n;i++) cin >>a[i];
	sort(a+1,a+1+n);
	a[n+1]=mod;
	f[0][0][0][0]=1;
	for (int i=1;i<=n;i++){
		for (int j=1;j<=i;j++) {
			for (int k=0;k<=K;k++) {
				for (int m=0;m<=2;m++) {
					int d=(2*j-m)*(a[i+1]-a[i]);
					if (d>k||i+j+1-m>n) continue;
					f[i][j][k][m]=f[i-1][j-1][k-d][m]+f[i-1][j][k-d][m]*(2*j-m);
					if (m) {
						f[i][j][k][m]+=f[i-1][j-1][k-d][m-1]*(3-m);
						if (m==1) f[i][j][k][m]+=f[i-1][j][k-d][m-1]*(2*j);
						else if (m==2) {
							if (i==n) f[i][j][k][m]+=f[i-1][j][k-d][m-1];
							else f[i][j][k][m]+=f[i-1][j][k-d][m-1]*(j-1);
						}
					}
					if (m==2) {
						if (i==n) f[i][j][k][m]+=f[i-1][j+1][k-d][m];
						else f[i][j][k][m]+=f[i-1][j+1][k-d][m]*(j*(j-1));
					}
					else if (m==1) f[i][j][k][m]+=f[i-1][j+1][k-d][m]*(j*j);
					else f[i][j][k][m]+=f[i-1][j+1][k-d][m]*(j*(j+1));
					f[i][j][k][m]%=mod;
				}
			}
		}
	}
	ll res=0;
	for (int i=0;i<=K;i++) (res+=f[n][1][i][2])%=mod;
	cout<<res;
	end=clock();	
	ld dur=(ld)(end-start)/(ld)(CLOCKS_PER_SEC)*(1000.0L);
	cerr<<dur;
    return 0;
}

• Subtask 1: 0 / 5
• Subtask 2: 15 / 15
• Subtask 3: 0 / 80